Documentation return type fixes