re-adding import to setup.py