Fixed bug I just introduced - grrrrr