Fix error in tests.