Fixed an issue I made in my last commit.