Fix up pylint errors