Fix bug
Merge pull request #49 from michaelnflx/hotfix

Fix bug