Fixing error with running the model