Fix setup.py
Merge pull request #1 from WyseNynja/patch-1

Fix wrong import in setup.py.