Fix pylint issues