Fixed bug I just introduced