1 files changed, 17 insertions, 0 deletions
diff --git a/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.Updater/MainWindow.xaml.cs b/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.Updater/MainWindow.xaml.cs
index e601b18f2..818ea7e41 100644
--- a/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.Updater/MainWindow.xaml.cs
+++ b/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.Updater/MainWindow.xaml.cs
@@ -49,6 +49,7 @@ namespace Tango.MachineStudio.Updater
             EnsureMachineStudioIsDown();
             //RemoveOldDLLFiles();
             ReplaceFiles();
+            PostProcessing();
             StartMachineStudio();
             Environment.Exit(0);
         }
@@ -176,5 +177,21 @@ namespace Tango.MachineStudio.Updater
         {
             MessageBox.Show(error, "Machine Studio Update", MessageBoxButton.OK, MessageBoxImage.Error);
         }
+
+        private void PostProcessing()
+        {
+            //Remove DB and ColorLab modules.
+            TryDelete(Path.Combine(_appPath, "Tango.MachineStudio.DB.dll"));
+            TryDelete(Path.Combine(_appPath, "Tango.MachineStudio.ColorLab.dll"));
+        }
+
+        private void TryDelete(String path)
+        {
+            try
+            {
+                File.Delete(path);
+            }
+            catch { }
+        }
     }
 }