Use this new function in places where it simplifies the code.
Signed-off-by: NSimon Glass <sjg@chromium.org>