Add test for #3
Fix test for #3
Merge pull request #4 from mjschultz/fix-3

Fix 3