Fix return type documentation