Fixed unit test breaking commit.