Fix the infinite loop that's causing the program to crash!