Fixing a pylint warning