Fix stupid bug I introduced