diff --git a/extended/code_update/host_code_update.robot b/extended/code_update/host_code_update.robot
index 473019a..3b4bd61 100644
--- a/extended/code_update/host_code_update.robot
+++ b/extended/code_update/host_code_update.robot
@@ -268,6 +268,8 @@
     Run Keyword If  'true' == '${DELETE_OLD_GUARD_FILE}'  BMC Execute Command
     ...  rm -f /var/lib/phosphor-software-manager/pnor/prsv/GUARD
 
+    BMC Execute Command  rm -rf /tmp/images/*
+    BMC Execute Command  systemctl restart xyz.openbmc_project.Software.BMC.Updater.service xyz.openbmc_project.Software.Version.service xyz.openbmc_project.Software.Download.service
 
 Upload And Activate Multiple BMC Images Setup
     [Documentation]  Check that the ALTERNATE_FILE_PATH variable is set.