Remove bug in unit test.