The original idea came from pHilipp, and this makes the code looks
more consistent.
Signed-off-by: NEric Miao <eric.miao@marvell.com>
Signed-off-by: NMark Brown <broonie@opensource.wolfsonmicro.com>