Bugfix: Wrong import in setup.py corrected