Fix style of for statement