Fix unit test that was broken with last commit.