Fix a bug I introduced