Fix some pylint issues