Updated the correct doc block for return type