Fix install via setup.py
Merge pull request #1 from Karmak23/master

Fix installation via setup.py