Fixed flake8 error that was causing CI Fails