From 8ad15559189d06bb70bfdf13c05c174ea97e3db8 Mon Sep 17 00:00:00 2001
From: schwiti6190 <58079399+schwiti6190@users.noreply.github.com>
Date: Sat, 26 Mar 2022 12:24:47 +0100
Subject: [PATCH] Possible animation fix for baleloaders mp

---
 .../ai/controllers/BaleLoaderController.lua   | 20 ++++---------------
 1 file changed, 4 insertions(+), 16 deletions(-)

diff --git a/scripts/ai/controllers/BaleLoaderController.lua b/scripts/ai/controllers/BaleLoaderController.lua
index 5311b1583..a2577b552 100644
--- a/scripts/ai/controllers/BaleLoaderController.lua
+++ b/scripts/ai/controllers/BaleLoaderController.lua
@@ -76,20 +76,8 @@ function BaleLoaderController:isFuelSaveAllowed()
     return false
 end
 
-BaleLoader.onAIImplementStart = Utils.overwrittenFunction(BaleLoader.onAIImplementStart,
-        function(self, superFunc)
-            if superFunc ~= nil then superFunc(self) end
-            if not g_server then 
-                return
-            end
-            self:doStateChange(BaleLoader.CHANGE_MOVE_TO_WORK);
-        end)
-
-local baleLoaderRegisterEventListeners = function(vehicleType)
-    print('## Courseplay: Registering event listeners for bale loader')
-    SpecializationUtil.registerEventListener(vehicleType, "onAIImplementStart", BaleLoader)
+function BaleLoaderController:onStart()
+    if not self.baleLoaderSpec.isInWorkPosition then
+        self.baleLoader:doStateChange(BaleLoader.CHANGE_BUTTON_WORK_TRANSPORT)
+    end
 end
-
-print('## Courseplay: Appending event listener for bale loaders')
-BaleLoader.registerEventListeners = Utils.appendedFunction(BaleLoader.registerEventListeners, baleLoaderRegisterEventListeners)
-