Fix pylint complaints