Fix functions in test.