Fix install
Merge pull request #21 from zoic21/master

Fix install