fix

ajkroeg · Sep 16, 2023 · 4fc2b02 · 4fc2b02
1 parent bbcb9c6
commit 4fc2b02
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/TrainingMissions.cs b/TrainingMissions.cs
@@ -121,8 +121,14 @@ public static void Postfix(LanceConfiguratorPanel __instance)
 //                    ModState.IsTrainingMission = true;
                     foreach (var slot in lanceLoadoutSlots)
                     {
-                        ModState.deployedMechs.Add(slot.SelectedMech.MechDef);
-                        ModInit.modLog.LogMessage($"Adding {slot.SelectedMech.MechDef.Name} to deployedMechs for restore");
+                        foreach (var kvp in new Dictionary<int, MechDef>(__instance.Sim.ActiveMechs))
+                        {
+                            if (slot.SelectedMech.MechDef.GUID == kvp.Value.GUID)
+                            {
+                                ModState.deployedMechs.Add(slot.SelectedMech.MechDef);
+                                ModInit.modLog.LogMessage($"Adding {slot.SelectedMech.MechDef.Name} to deployedMechs for restore");
+                            } 
+                        }
                     }
                 }
             }

diff --git a/TrainingMissions.csproj b/TrainingMissions.csproj
@@ -34,8 +34,8 @@
   <PropertyGroup>
     <!-- avoids IgnoresAccessChecksToAttribute warnings -->
     <PublicizerRuntimeStrategies>Unsafe</PublicizerRuntimeStrategies>
-    <AssemblyVersion>1.1.0.0</AssemblyVersion>
-    <FileVersion>1.1.0.0</FileVersion>
+    <AssemblyVersion>1.1.0.1</AssemblyVersion>
+    <FileVersion>1.1.0.1</FileVersion>
   </PropertyGroup>
   <ItemGroup>
     <PackageReference Include="Krafs.Publicizer" Version="2.2.1" />