Fixed infinite loop bug I think