CODE: Fixed linting code