Fixed a bug I introduced (I think it was me)