Corrected a flake8 issue -- this line is imported for the purpose of re-exposing the name