1 files changed, 1 insertions, 0 deletions

diff --git a/src/WixToolset.WixBA/UpdateViewModel.cs b/src/WixToolset.WixBA/UpdateViewModel.cs
index cb6c0ad9..4a97afec 100644
--- a/src/WixToolset.WixBA/UpdateViewModel.cs
+++ b/src/WixToolset.WixBA/UpdateViewModel.cs
@@ -6,6 +6,7 @@ namespace WixToolset.WixBA
     using System.ComponentModel;
     using System.Windows.Input;
     using WixToolset.BootstrapperCore;
+    using WixToolset.Mba.Core;
 
     /// <summary>
     /// The states of the update view model.