Fix broken unit test
Merge pull request #4 from jwhitbeck/master

Fix broken unit test