diff --git a/lib/redfish_code_update_utils.robot b/lib/redfish_code_update_utils.robot
index 916540e..4395ef2 100644
--- a/lib/redfish_code_update_utils.robot
+++ b/lib/redfish_code_update_utils.robot
@@ -341,6 +341,26 @@
     [Return]  ${task_inv}
 
 
+Get Task Inventory
+    [Documentation]  Return task inventory.
+    [Arguments]  ${task_info}
+
+    # Description of argument(s):
+    # task_info    Task information.
+
+    # Task information.
+    # @odata.id: /redfish/v1/TaskService/Tasks/1
+    # Id: 1
+    # TaskState: Starting
+    # TaskStatus: OK
+
+    ${task_payload}=  Redfish.Get Properties   ${task_info['@odata.id']}
+
+    ${task_inv}=  Create Initiated Task State Dict  ${task_payload}
+
+    [Return]  ${task_inv}
+
+
 Match Target URI
     [Documentation]  Match target uri from task list.
     [Arguments]  ${task_list}  ${target_uri}
diff --git a/lib/rest_client.robot b/lib/rest_client.robot
index c82087f..2ebbd14 100644
--- a/lib/rest_client.robot
+++ b/lib/rest_client.robot
@@ -372,7 +372,7 @@
     Valid Value  ret.status_code  ${valid_status_codes}
     Delete All Sessions
 
-    [Return]  ${ret.status_code}
+    [Return]  ${ret}
 
 
 Redfish Login