Slightly rework return value handling, no functional changes.
Signed-off-by: NJeffy Chen <jeffy.chen@rock-chips.com>
Signed-off-by: NMark Brown <broonie@kernel.org>