Fix linting for CI