Fixes/infinite loop