diff --git a/Moose Development/Moose/Functional/AIBalancer.lua b/Moose Development/Moose/Functional/AIBalancer.lua
index a2cd679e6..8d6e4b47e 100644
--- a/Moose Development/Moose/Functional/AIBalancer.lua	
+++ b/Moose Development/Moose/Functional/AIBalancer.lua	
@@ -138,7 +138,7 @@ end
 -- @param Core.Set#SET_GROUP SetGroup
 -- @param #string ClientName
 -- @param Wrapper.Group#GROUP AIGroup
-function AIBALANCER:onenterSpawning( SetGroup, ClientName )
+function AIBALANCER:onenterSpawning( SetGroup, Event, From, To, ClientName )
 
   -- OK, Spawn a new group from the default SpawnAI object provided.
   local AIGroup = self.SpawnAI:Spawn()
@@ -151,7 +151,7 @@ end
 --- @param #AIBALANCER self
 -- @param Core.Set#SET_GROUP SetGroup
 -- @param Wrapper.Group#GROUP AIGroup
-function AIBALANCER:onenterDestroying( SetGroup, AIGroup )
+function AIBALANCER:onenterDestroying( SetGroup, Event, From, To, AIGroup )
 
   AIGroup:Destroy()
 end
@@ -159,7 +159,7 @@ end
 --- @param #AIBALANCER self
 -- @param Core.Set#SET_GROUP SetGroup
 -- @param Wrapper.Group#GROUP AIGroup
-function AIBALANCER:onenterReturning( SetGroup, AIGroup )
+function AIBALANCER:onenterReturning( SetGroup, Event, From, To, AIGroup )
 
     local AIGroupTemplate = AIGroup:GetTemplate()
     if self.ToHomeAirbase == true then
diff --git a/Moose Test Missions/AI Balancer/AIBALANCER - One coalition InitCleanUp API test/AIBALANCER - One coalition InitCleanUp API test.miz b/Moose Test Missions/AI Balancer/AIBALANCER - One coalition InitCleanUp API test/AIBALANCER - One coalition InitCleanUp API test.miz
index 6aa474729..b0fc85ea2 100644
Binary files a/Moose Test Missions/AI Balancer/AIBALANCER - One coalition InitCleanUp API test/AIBALANCER - One coalition InitCleanUp API test.miz and b/Moose Test Missions/AI Balancer/AIBALANCER - One coalition InitCleanUp API test/AIBALANCER - One coalition InitCleanUp API test.miz differ