fixed infinite loop :no_good: