1 files changed, 9 insertions, 0 deletions

diff --git a/src/test/burn/WixToolset.WixBA/InstallationViewModel.cs b/src/test/burn/WixToolset.WixBA/InstallationViewModel.cs
index 14ea561f..1846d51b 100644
--- a/src/test/burn/WixToolset.WixBA/InstallationViewModel.cs
+++ b/src/test/burn/WixToolset.WixBA/InstallationViewModel.cs
@@ -85,6 +85,7 @@ namespace WixToolset.WixBA
             WixBA.Model.Bootstrapper.ExecutePackageComplete += this.ExecutePackageComplete;
             WixBA.Model.Bootstrapper.Error += this.ExecuteError;
             WixBA.Model.Bootstrapper.ApplyComplete += this.ApplyComplete;
+            WixBA.Model.Bootstrapper.SetUpdateComplete += this.SetUpdateComplete;
         }
 
         void RootPropertyChanged(object sender, PropertyChangedEventArgs e)
@@ -409,6 +410,14 @@ namespace WixToolset.WixBA
             }
         }
 
+        private void SetUpdateComplete(object sender, SetUpdateCompleteEventArgs e)
+        {
+            if (!String.IsNullOrEmpty(e.NewPackageId) && !WixBA.Model.BAManifest.Bundle.Packages.ContainsKey(e.NewPackageId))
+            {
+                WixBA.Model.BAManifest.Bundle.AddUpdateBundleAsPackage(e);
+            }
+        }
+
         private void DetectComplete(object sender, DetectCompleteEventArgs e)
         {
             // Parse the command line string before any planning.