null check and 1 less patch

eDonnes124 · Apr 27, 2024 · a76edfc · a76edfc
1 parent 91a157f
commit a76edfc
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 13 deletions.
diff --git a/source/Patches/CrewmateRoles/InvestigatorMod/AddPrints.cs b/source/Patches/CrewmateRoles/InvestigatorMod/AddPrints.cs
@@ -22,7 +22,7 @@ private static Vector2 Position(PlayerControl player)
 
         public static void Postfix()
         {
-            if (!GameManager.Instance.GameHasStarted || !PlayerControl.LocalPlayer.Is(RoleEnum.Investigator)) return;
+            if ((GameManager.Instance && !GameManager.Instance.GameHasStarted) || !PlayerControl.LocalPlayer.Is(RoleEnum.Investigator)) return;
             if (MeetingHud.Instance) return;
 
             // New Footprint

diff --git a/source/Patches/CrewmateRoles/InvestigatorMod/EndGame.cs b/source/Patches/CrewmateRoles/InvestigatorMod/EndGame.cs
@@ -11,22 +11,13 @@ public static void Reset()
         }
 
         [HarmonyPatch(typeof(AmongUsClient), nameof(AmongUsClient.ExitGame))]
-        public static class EndGamePatch
-        {
-            public static void Prefix(AmongUsClient __instance)
-            {
-                Reset();
-            }
-        }
-
         [HarmonyPatch(typeof(EndGameManager), nameof(EndGameManager.Start))]
-        public static class EndGameManagerPatch
+
+        public static class EndGamePatch
         {
-            public static bool Prefix(EndGameManager __instance)
+            public static void Prefix()
             {
                 Reset();
-
-                return true;
             }
         }
     }