Fix little type error that I introduced