Revert "fix #8"

This reverts commit 9be72bae2c6352bd2b64e6a55754859c461a220f.
fix #9
Merge pull request #11 from ddavison/fix-9

Fix 9