Missing import in setup.py