Change place of code [minor]