Fix infinite loop :)