This is more idiomatic and interacts better with deferred probing.
Signed-off-by: NMark Brown <broonie@linaro.org>