Fixed a bug I just introduced