Correct type fix