Move the code to the right place