diff --git a/extended/code_update/host_code_update.robot b/extended/code_update/host_code_update.robot
index df728b5..6feafe2 100644
--- a/extended/code_update/host_code_update.robot
+++ b/extended/code_update/host_code_update.robot
@@ -62,6 +62,7 @@
     Start SOL Console Logging
     Delete Error Logs
     REST Power On
+    Verify Running Host Image  ${IMAGE_FILE_PATH}
 
 
 Host Image Priority Attribute Test