diff --git a/extended/code_update/test_field_mode.robot b/extended/code_update/test_field_mode.robot
index 91b5cd9..42bdcb3 100644
--- a/extended/code_update/test_field_mode.robot
+++ b/extended/code_update/test_field_mode.robot
@@ -30,7 +30,7 @@
     #         ],
     #         [
     #             "functional",
-    #             "software_version",
+    #             "functional",
     #             "/xyz/openbmc_project/software/fcf8e182"
     #         ]
     #     ]
diff --git a/lib/code_update_utils.robot b/lib/code_update_utils.robot
index 978ddcd..fc06b89 100644
--- a/lib/code_update_utils.robot
+++ b/lib/code_update_utils.robot
@@ -399,7 +399,7 @@
     #         ],
     #         [
     #             "functional",
-    #             "software_version",
+    #             "functional",
     #             "/xyz/openbmc_project/software/fcf8e182"
     #         ]
     #     ]