1 files changed, 10 insertions, 4 deletions
diff --git a/Software/Visual_Studio/Azure/Tango.AzureUtils.UI/ViewModels/EnvironmentFirmwareUpgradeViewVM.cs b/Software/Visual_Studio/Azure/Tango.AzureUtils.UI/ViewModels/EnvironmentFirmwareUpgradeViewVM.cs
index 54576cda7..964febbd8 100644
--- a/Software/Visual_Studio/Azure/Tango.AzureUtils.UI/ViewModels/EnvironmentFirmwareUpgradeViewVM.cs
+++ b/Software/Visual_Studio/Azure/Tango.AzureUtils.UI/ViewModels/EnvironmentFirmwareUpgradeViewVM.cs
@@ -94,10 +94,6 @@ namespace Tango.AzureUtils.UI.ViewModels
                 IsFree = false;
 
                 await _firmwareManager.InjectFirmwarePackage(SelectedDeploymentSlot, FilePath);
-
-                var old = SelectedDeploymentSlot;
-                SelectedDeploymentSlot = null;
-                SelectedDeploymentSlot = old;
             }
             catch (Exception ex)
             {
@@ -105,8 +101,18 @@ namespace Tango.AzureUtils.UI.ViewModels
             }
             finally
             {
+                RequireRefresh();
                 IsFree = true;
             }
         }
+
+        protected override void OnRefreshRequired()
+        {
+            base.OnRefreshRequired();
+
+            var old = SelectedDeploymentSlot;
+            SelectedDeploymentSlot = null;
+            SelectedDeploymentSlot = old;
+        }
     }
 }