Fix setup.py imports.