Fix infinite loop I caused