Fixed slight issue that I created at 4am