Moving a couple of lines around allows us to shrink the code a bit
while keeping the same functionality.
Signed-off-by: NBartosz Golaszewski <bgolaszewski@baylibre.com>
Signed-off-by: NLinus Walleij <linus.walleij@linaro.org>