fix for bad import in setup.py