diff --git a/activation.cpp b/activation.cpp
index 79b5b10..c82e297 100644
--- a/activation.cpp
+++ b/activation.cpp
@@ -418,11 +418,11 @@
         // unsubscribe to systemd signals
         unsubscribeFromSystemdSignals();
 
-        // Remove version object from image manager
-        deleteImageManagerObject();
-
         if (newStateResult == "done")
         {
+            // Remove version object from image manager
+            deleteImageManagerObject();
+
             // Set activation progress to 100
             activationProgress->progress(100);