diff --git a/activation.cpp b/activation.cpp
index 6029f33..c90b537 100644
--- a/activation.cpp
+++ b/activation.cpp
@@ -418,9 +418,6 @@
 
         if (newStateResult == "done")
         {
-            // Remove version object from image manager
-            deleteImageManagerObject();
-
             // Set activation progress to 100
             activationProgress->progress(100);
 
diff --git a/item_updater.cpp b/item_updater.cpp
index b6464f6..fe9c808 100644
--- a/item_updater.cpp
+++ b/item_updater.cpp
@@ -363,6 +363,7 @@
     else
     {
         removeAssociations(iteratorActivations->second->path);
+        iteratorActivations->second->deleteImageManagerObject();
         this->activations.erase(entryId);
     }
     ItemUpdater::resetUbootEnvVars();