Fix return type in test helper.