Added bugfix.
Merge pull request #111 from Zanarama/master

Added bugfix.