setup.py:

Added missing import.