Fix broken unit test
Merge pull request #481 from nicks/nick-tests

Fix broken unit test