PyLint fix for CI