Fixed unit test broken by the last commit.