Fixing the for_loop test