Fix Pylint errors