Fix return value type.