add the import I missed
correction on setup.py