1 files changed, 7 insertions, 3 deletions
diff --git a/Software/Visual_Studio/PPC/Tango.PPC.UI/ViewModels/MachineSetupViewVM.cs b/Software/Visual_Studio/PPC/Tango.PPC.UI/ViewModels/MachineSetupViewVM.cs
index 514a517ff..1bd1e1eea 100644
--- a/Software/Visual_Studio/PPC/Tango.PPC.UI/ViewModels/MachineSetupViewVM.cs
+++ b/Software/Visual_Studio/PPC/Tango.PPC.UI/ViewModels/MachineSetupViewVM.cs
@@ -18,6 +18,7 @@ using Tango.PPC.Common.Connection;
 using Tango.PPC.Common.MachineSetup;
 using Tango.PPC.Common.Navigation;
 using Tango.PPC.Common.OS;
+using Tango.PPC.Common.Web;
 using Tango.PPC.UI.ViewsContracts;
 using Tango.Settings;
 using Tango.SharedUI.Helpers;
@@ -60,6 +61,7 @@ namespace Tango.PPC.UI.ViewModels
         private MachineSetupResult _setup_result;
         private IOperationSystemManager _operationSystemManager;
         private IPPCApplicationManager _appManager;
+        private PPCWebClient _ppcWebClient;
 
         #region Properties
 
@@ -192,8 +194,9 @@ namespace Tango.PPC.UI.ViewModels
         /// </summary>
         /// <param name="applicationManager">The application manager.</param>
         /// <param name="machineSetupManager">The machine setup manager.</param>
-        public MachineSetupViewVM(IPPCApplicationManager applicationManager, IMachineSetupManager machineSetupManager, IOperationSystemManager operationSystemManager)
+        public MachineSetupViewVM(PPCWebClient ppcWebClient, IPPCApplicationManager applicationManager, IMachineSetupManager machineSetupManager, IOperationSystemManager operationSystemManager)
         {
+            _ppcWebClient = ppcWebClient;
             _appManager = applicationManager;
             MachineSetupManager = machineSetupManager;
 
@@ -210,7 +213,7 @@ namespace Tango.PPC.UI.ViewModels
             InstallCommand = new RelayCommand(Install);
             RestartCommand = new RelayCommand(() => { NavigateTo(MachineSetupView.WelcomeView); });
             TimeZoneSelectedCommand = new RelayCommand(() => { NavigateTo(MachineSetupView.EnvironmentView); });
-            EnvironmentSelectedCommand = new RelayCommand(() => 
+            EnvironmentSelectedCommand = new RelayCommand(() =>
             {
                 NavigateTo(MachineSetupView.SetupWelcomeView);
                 Settings.DeploymentSlot = DeploymentSlot;
@@ -348,8 +351,9 @@ namespace Tango.PPC.UI.ViewModels
 
             try
             {
+                _ppcWebClient.Environment = DeploymentSlot;
                 await _operationSystemManager.ChangeTimeZone(SelectedTimeZone);
-                _setup_result = await MachineSetupManager.Setup(SerialNumber, HostAddress);
+                _setup_result = await MachineSetupManager.Setup(SerialNumber);
                 State = MachineSetupStates.Completed;
                 LogManager.Log("Machine setup completed.");
                 await NavigateTo(MachineSetupView.SetupCompletedView);