Listen to onLaunch rather than onStageActivated

severedsolo · May 21, 2017 · c3bc151 · c3bc151
1 parent 8dbd9b7
commit c3bc151
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/UPFM/Untitled Part Failure Mod/BaseFailureModule.cs b/UPFM/Untitled Part Failure Mod/BaseFailureModule.cs
@@ -25,13 +25,13 @@ private void Start()
             part.AddModule("DontRecoverMe");
             ScrapYardEvents.OnSYTrackerUpdated.Add(OnSYTrackerUpdated);
             Initialise();
-            GameEvents.onStageActivate.Add(onStageActivate);
+            GameEvents.onLaunch.Add(onLaunch);
         }
 
-        private void onStageActivate(int data)
+        private void onLaunch(EventReport data)
         {
             PartModule dontRecover = part.FindModuleImplementing<DontRecoverMe>();
-            if(dontRecover == null) return;
+            if (dontRecover == null) return;
             part.RemoveModule(dontRecover);
             Debug.Log("[UPFM]: " + part.name + "marked as recoverable");
         }
@@ -125,7 +125,7 @@ public void RepairChecks()
 
         private void OnDestroy()
         {
-            GameEvents.onStageActivate.Remove(onStageActivate);
+            GameEvents.onLaunch.Remove(onLaunch);
             if (ScrapYardEvents.OnSYTrackerUpdated == null) return;
             ScrapYardEvents.OnSYTrackerUpdated.Remove(OnSYTrackerUpdated);
         }