Fixed infinite loop. Whoops