trying to fix the for loop test