Moved some code (1 line)