Fix doc block return type