Fix import line for setup.py