Fix pylint warnings