Minor pylint fix