Corrected a flake8 check failure