Add test for FOR loop issue