Patch up unit test I broke in pull request 7