Fix pylint errors