Fix #2
Merge pull request #3 from rafaels/fix_bug

Fix #2