Fix critical bug
Added fix to the fix
Merge pull request #9 from PeterBell/fix-bug-1

Fix critical bug