Fix error in unit test.