fixed the model that was causing errors