Fix return type.