diff --git a/binaries/data/mods/public/simulation/ai/common-api/baseAI.js b/binaries/data/mods/public/simulation/ai/common-api/baseAI.js
index 7cb2fba594..9efa2e81ea 100644
--- a/binaries/data/mods/public/simulation/ai/common-api/baseAI.js
+++ b/binaries/data/mods/public/simulation/ai/common-api/baseAI.js
@@ -20,7 +20,6 @@ BaseAI.prototype.Serialize = function()
  */
 BaseAI.prototype.Deserialize = function(data, sharedScript)
 {
-	this.isDeserialized = true;
 };
 
 BaseAI.prototype.Init = function(state, playerID, sharedAI)
@@ -47,12 +46,6 @@ BaseAI.prototype.HandleMessage = function(state, playerID, sharedAI)
 {
 	PlayerID = playerID;
 	this.territoryMap = sharedAI.territoryMap;
-
-	if (this.isDeserialized)
-	{
-		this.Init(state, playerID, sharedAI);
-		this.isDeserialized = false;
-	}
 	this.OnUpdate(sharedAI);
 };
 
diff --git a/binaries/data/mods/public/simulation/ai/petra/_petrabot.js b/binaries/data/mods/public/simulation/ai/petra/_petrabot.js
index 00ed5342bf..3c8f161553 100644
--- a/binaries/data/mods/public/simulation/ai/petra/_petrabot.js
+++ b/binaries/data/mods/public/simulation/ai/petra/_petrabot.js
@@ -91,6 +91,9 @@ PetraBot.prototype.CustomInit = function(gameState)
 
 PetraBot.prototype.OnUpdate = function(sharedScript)
 {
+	if (this.isDeserialized)
+		this.Init(state, playerID, sharedAI);
+
 	if (this.gameFinished || this.gameState.playerData.state == "defeated")
 		return;