Move a return statement to the correct position.