Fix class name for the model item