Fixup menu code

2010-11-17 09:47:53 +01:00
parent cc12877e04
commit 1ffdf9b2cb
4 changed files with 13 additions and 13 deletions
--- a/gui/builtinContextMenus/moduleAmmoPicker.py
+++ b/gui/builtinContextMenus/moduleAmmoPicker.py
@@ -14,10 +14,9 @@ class ModuleAmmoPicker(ContextMenu):
        if self.mainFrame.getActiveFit() is None or context not in ("module", "projectedModule"):
            return False

-        if context == "module":
-            modules = self.mainFrame.getFittingView().getSelectedMods()
-        else:
-            modules = (selection[0],)
+        modules = selection if context == "module" else (selection[0],)
+
+
        validCharges = None
        for mod in modules:
            currCharges = mod.getValidCharges()