fix broken unit test