Small bug when running after the first commit

These should fix it