Fix endless loop :/