Fixed what was broken in my code