Updated broken unit test.