Fix Pylint warnings