fix #7
Merge pull request #8 from ddavison/fix-7

fix #7