Pylint fix for new test