Fix pylint failure