fixed a broken unit test