Fixing linting bug