Fixing linting of models