Fixed a bug I made last commit :/