Fix import error in setup.py
Fixes #17