Moved some code around

Signed-off-by: Irmo van den Berge <bergerkiller@gmail.com>