fixing the models