Fix error in test.