added missing import to setup.py