From 66c18834da7474dd921f09b5781a73b9bd7e1089 Mon Sep 17 00:00:00 2001
From: KaydemonLP <35258821+KaydemonLP@users.noreply.github.com>
Date: Wed, 10 Dec 2025 15:47:15 +0100
Subject: [PATCH] Add vehicle idle state machine which is now necessary

---
 HenryMod/Modules/Prefabs.cs | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/HenryMod/Modules/Prefabs.cs b/HenryMod/Modules/Prefabs.cs
index fd5ab7b..c2932a6 100644
--- a/HenryMod/Modules/Prefabs.cs
+++ b/HenryMod/Modules/Prefabs.cs
@@ -634,6 +634,14 @@ public static void ClearEntityStateMachines(GameObject bodyPrefab)
             if (setStateOnHurt)
             {
                 setStateOnHurt.idleStateMachine = Array.Empty<EntityStateMachine>();
+            }
+			
+			// Ideally this should be changed to your weapon entity state machine
+			// Otherwise it lets you continue attacking during ziprails and similar
+            CharacterBody characterBody = bodyPrefab.GetComponent<CharacterBody>();
+            if (characterBody)
+            {
+                characterBody.vehicleIdleStateMachine = Array.Empty<EntityStateMachine>();
             }
         }