small bugfix
Merge pull request #9 from marius0/master

Fixing a small bug