Fix bug
Merge pull request #211 from jiikko/bug-fix

Fix bug