fix a broken unit test