Fix unit test
Merge pull request #14 from sarahholt/master

Fix unit test