a little type fix