Fixed broken unit test.