Import missing in setup.py