Fix linting warnings