diff --git a/gui/builtinAdditionPanes/implantView.py b/gui/builtinAdditionPanes/implantView.py index 1dfea89d3..2d0c144b1 100644 --- a/gui/builtinAdditionPanes/implantView.py +++ b/gui/builtinAdditionPanes/implantView.py @@ -261,7 +261,7 @@ class ImplantDisplay(d.Display): sMkt = Market.getInstance() sourceContext = "implantItem" if fit.implantSource == ImplantLocation.FIT else "implantItemChar" itemContext = sMkt.getCategoryByItem(implant.item).name - fullContext = ((sourceContext, itemContext), ("implantView",)) + fullContext = ((sourceContext, itemContext), ("implantView", itemContext)) menu = ContextMenu.getMenu((implant,), *fullContext) elif sel == -1 and fit.implantSource == ImplantLocation.FIT: fitID = self.mainFrame.getActiveFit() diff --git a/gui/fitCommands/calc/implant/add.py b/gui/fitCommands/calc/implant/add.py index 06c864ccb..f21929df9 100644 --- a/gui/fitCommands/calc/implant/add.py +++ b/gui/fitCommands/calc/implant/add.py @@ -58,8 +58,12 @@ class CalcAddImplantCommand(wx.Command): def Undo(self): pyfalog.debug('Undo addition of implant {} to fit {}'.format(self.newImplantInfo, self.fitID)) if self.oldImplantInfo is not None and self.oldPosition is not None: - cmd = CalcAddImplantCommand(fitID=self.fitID, implantInfo=self.oldImplantInfo, position=self.oldPosition) + cmd = CalcAddImplantCommand( + fitID=self.fitID, + implantInfo=self.oldImplantInfo, + position=self.oldPosition, + commit=self.commit) return cmd.Do() from .remove import CalcRemoveImplantCommand - cmd = CalcRemoveImplantCommand(fitID=self.fitID, position=self.newPosition) + cmd = CalcRemoveImplantCommand(fitID=self.fitID, position=self.newPosition, commit=self.commit) return cmd.Do() diff --git a/gui/fitCommands/calc/implant/remove.py b/gui/fitCommands/calc/implant/remove.py index c7c3c546c..508eab50d 100644 --- a/gui/fitCommands/calc/implant/remove.py +++ b/gui/fitCommands/calc/implant/remove.py @@ -31,5 +31,9 @@ class CalcRemoveImplantCommand(wx.Command): def Undo(self): pyfalog.debug('Undoing removal of implant {} on fit {}'.format(self.savedImplantInfo, self.fitID)) from .add import CalcAddImplantCommand - cmd = CalcAddImplantCommand(fitID=self.fitID, implantInfo=self.savedImplantInfo, position=self.position, commit=self.commit) + cmd = CalcAddImplantCommand( + fitID=self.fitID, + implantInfo=self.savedImplantInfo, + position=self.position, + commit=self.commit) return cmd.Do()