Fixed a bug I caused in the last commit