Fix return tests.