diff --git a/gui/builtinContextMenus/droneRemoveStack.py b/gui/builtinContextMenus/droneRemoveStack.py index e465f4b2d..c3063ca52 100644 --- a/gui/builtinContextMenus/droneRemoveStack.py +++ b/gui/builtinContextMenus/droneRemoveStack.py @@ -18,10 +18,10 @@ class ItemRemove(ContextMenu): srcContext = fullContext[0] sFit = service.Fit.getInstance() fitID = self.mainFrame.getActiveFit() - sFit = sFit.getFit(fitID) + fit = sFit.getFit(fitID) - idx = sFit.drones.index(selection[0]) - sFit.removeDrone(fitID, idx, numDronesToRemove=sFit.drones[idx].amount) + idx = fit.drones.index(selection[0]) + sFit.removeDrone(fitID, idx, numDronesToRemove=fit.drones[idx].amount) wx.PostEvent(self.mainFrame, GE.FitChanged(fitID=fitID)) diff --git a/gui/builtinContextMenus/itemRemove.py b/gui/builtinContextMenus/itemRemove.py index c291a1d37..a54c415b7 100644 --- a/gui/builtinContextMenus/itemRemove.py +++ b/gui/builtinContextMenus/itemRemove.py @@ -19,20 +19,20 @@ class ItemRemove(ContextMenu): srcContext = fullContext[0] sFit = service.Fit.getInstance() fitID = self.mainFrame.getActiveFit() - sFit = sFit.getFit(fitID) + fit = sFit.getFit(fitID) if srcContext == "fittingModule": for module in selection: if module is not None: - sFit.removeModule(fitID,sFit.modules.index(module)) + sFit.removeModule(fitID,fit.modules.index(module)) elif srcContext in ("fittingCharge" , "projectedCharge"): sFit.setAmmo(fitID, None, selection) elif srcContext == "droneItem": - sFit.removeDrone(fitID, sFit.drones.index(selection[0])) + sFit.removeDrone(fitID, fit.drones.index(selection[0])) elif srcContext == "implantItem": - sFit.removeImplant(fitID, sFit.implants.index(selection[0])) + sFit.removeImplant(fitID, fit.implants.index(selection[0])) elif srcContext == "boosterItem": - sFit.removeBooster(fitID, sFit.boosters.index(selection[0])) + sFit.removeBooster(fitID, fit.boosters.index(selection[0])) else: sFit.removeProjected(fitID, selection[0])