Fixing the bug I introduced!