Refix segfault/infinite loop