fix up the broken unittest for last commit.