Fix bug in setup.py

@robertkluin-wf