Modified setup script. Now works as expected with pip install