Fix bug I introduced