Minor change to for loop