From 21a071ea406e857a7b96cb401c30586ea8df0de8 Mon Sep 17 00:00:00 2001
From: blitzmann <holmes.ryan.90@gmail.com>
Date: Mon, 8 May 2017 22:34:03 -0400
Subject: [PATCH] Add "remove" context menu to command fits

---
 gui/builtinContextMenus/itemRemove.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/gui/builtinContextMenus/itemRemove.py b/gui/builtinContextMenus/itemRemove.py
index 6ad128220..e5bb32d0e 100644
--- a/gui/builtinContextMenus/itemRemove.py
+++ b/gui/builtinContextMenus/itemRemove.py
@@ -21,12 +21,14 @@ class ItemRemove(ContextMenu):
                               "boosterItem", "projectedModule",
                               "projectedCharge", "cargoItem",
                               "projectedFit", "projectedDrone",
-                              "fighterItem", "projectedFighter")
+                              "fighterItem", "projectedFighter",
+                              "commandFit")
 
     def getText(self, itmContext, selection):
         return "Remove {0}".format(itmContext if itmContext is not None else "Item")
 
     def activate(self, fullContext, selection, i):
+
         srcContext = fullContext[0]
         sFit = Fit.getInstance()
         fitID = self.mainFrame.getActiveFit()
@@ -48,8 +50,10 @@ class ItemRemove(ContextMenu):
             sFit.removeBooster(fitID, fit.boosters.index(selection[0]))
         elif srcContext == "cargoItem":
             sFit.removeCargo(fitID, fit.cargo.index(selection[0]))
-        else:
+        elif srcContext == "projectedFit":
             sFit.removeProjected(fitID, selection[0])
+        elif srcContext == "commandFit":
+            sFit.removeCommand(fitID, selection[0])
 
         wx.PostEvent(self.mainFrame, GE.FitChanged(fitID=fitID))