Working on PPC packages...

1 files changed, 10 insertions, 0 deletions

diff --git a/Software/Visual_Studio/PPC/Tango.PPC.UI/ViewModelLocator.cs b/Software/Visual_Studio/PPC/Tango.PPC.UI/ViewModelLocator.cs
index ed56f8a08..60bce4e20 100644
--- a/Software/Visual_Studio/PPC/Tango.PPC.UI/ViewModelLocator.cs
+++ b/Software/Visual_Studio/PPC/Tango.PPC.UI/ViewModelLocator.cs
@@ -74,6 +74,7 @@ namespace Tango.PPC.UI
             TangoIOC.Default.Unregister<IOperationSystemManager>();
             TangoIOC.Default.Unregister<PPCWebClient>();
             TangoIOC.Default.Unregister<IBackupManager>();
+            TangoIOC.Default.Unregister<IPackageRunner>();
 
             TangoIOC.Default.Register<PPCWebClient, PPCWebClient>(new PPCWebClient());
             TangoIOC.Default.Register<IDispatcherProvider, DefaultDispatcherProvider>(new DefaultDispatcherProvider(Application.Current.Dispatcher));
@@ -110,6 +111,7 @@ namespace Tango.PPC.UI
             TangoIOC.Default.Register<NoPermissionsViewVM>();
             TangoIOC.Default.Register<RestartingSystemViewVM>();
             TangoIOC.Default.Register<EmergencyViewVM>();
+            TangoIOC.Default.Register<RestartingViewVM>();
 
 
             TangoIOC.Default.GetInstance<IPPCApplicationManager>().ContentRendered += (_, __) =>
@@ -215,5 +217,13 @@ namespace Tango.PPC.UI
                 return TangoIOC.Default.GetInstance<EmergencyViewVM>();
             }
         }
+
+        public static RestartingViewVM RestartingViewVM
+        {
+            get
+            {
+                return TangoIOC.Default.GetInstance<RestartingViewVM>();
+            }
+        }
     }
 }
\ No newline at end of file