diff --git a/cmd/updatehub/main.go b/cmd/updatehub/main.go
index a2fac027..22bb1822 100644
--- a/cmd/updatehub/main.go
+++ b/cmd/updatehub/main.go
@@ -94,6 +94,7 @@ func main() {
 	}
 
 	settings.PersistentPollingSettings = runtimeSettings.PersistentPollingSettings
+	settings.PersistentUpdateSettings = runtimeSettings.PersistentUpdateSettings
 
 	log.Info("starting UpdateHub Agent")
 	log.Info("    version: ", gitversion)
diff --git a/updatehub/updatehub.go b/updatehub/updatehub.go
index 922f6f9b..129527ca 100644
--- a/updatehub/updatehub.go
+++ b/updatehub/updatehub.go
@@ -496,6 +496,7 @@ func (uh *UpdateHub) InstallUpdate(updateMetadata *metadata.UpdateMetadata, prog
 		}
 
 		log.Info("ActiveInactive activated: ", indexToInstall)
+		uh.Settings.UpgradeToInstallation = indexToInstall
 	}
 
 	log.Info("update installed successfully")