fix a bug I introduced