Fixed unit test broken by previous commit.