diff --git a/extended/code_update/code_update_utils.robot b/extended/code_update/code_update_utils.robot
index 715c8e2..6c33092 100644
--- a/extended/code_update/code_update_utils.robot
+++ b/extended/code_update/code_update_utils.robot
@@ -32,10 +32,9 @@
     ${data} =   create dictionary   data=@{img_path}
     ${resp}=    openbmc post request    ${BMC_UPD_METHOD}   data=${data}
     should be equal as strings   ${resp.status_code}   ${HTTP_OK}
-    ${content}=     To Json    ${resp.content}
-    should be equal as strings   ${content["data"]["filename"]}   /tmp/flashimg
 
     ${data}=      Read Properties     ${BMC_UPD_ATTR}
+    should be equal as strings   ${data["filename"]}   /tmp/flashimg
     should contain    ${data['status']}   to apply