Just a little type fix