Fixed a minor bug I just introduced