1 files changed, 6 insertions, 0 deletions
diff --git a/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.UI/ViewModels/FirmwareUpgradeViewVM.cs b/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.UI/ViewModels/FirmwareUpgradeViewVM.cs
index e45c29f73..7d2d8b401 100644
--- a/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.UI/ViewModels/FirmwareUpgradeViewVM.cs
+++ b/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.UI/ViewModels/FirmwareUpgradeViewVM.cs
@@ -202,11 +202,17 @@ namespace Tango.MachineStudio.UI.ViewModels
             {
                 IsFree = true;
                 CanClose = true;
+                _stream?.Dispose();
                 UpgradeError = ex.FlattenMessage();
                 CurrentPage = 3;
             }
         }
 
+        protected override bool CanOK()
+        {
+            return base.CanOK() && CanClose;
+        }
+
         private async void AbortUpgrade()
         {
             CanClose = true;