Fix minor type in the model