changed the model to fix the problem