diff --git a/AI/src/game/penguin.cpp b/AI/src/game/penguin.cpp
index 4d9d6c83c7d66423bfe91af8b99385ea4be33cff..0ccb4c9359cdabeed88aefcbf7921e259eeeeb5c 100644
--- a/AI/src/game/penguin.cpp
+++ b/AI/src/game/penguin.cpp
@@ -386,14 +386,11 @@ namespace game
 		json_state["bitboards"]["twofish"] = state.two_fish;
 		json_state["bitboards"]["threefish"] = state.three_fish;
 
-		json_state["penguins"]["red"][0] = state.p1_red;
-		json_state["penguins"]["red"][1] = state.p2_red;
-		json_state["penguins"]["red"][2] = state.p3_red;
-		json_state["penguins"]["red"][3] = state.p4_red;
-		json_state["penguins"]["blue"][0] = state.p1_blue;
-		json_state["penguins"]["blue"][1] = state.p2_blue;
-		json_state["penguins"]["blue"][2] = state.p3_blue;
-		json_state["penguins"]["blue"][3] = state.p4_blue;
+		for(int i = 0; i < 4; i++)
+		{
+			json_state["penguins"]["red"][i] = state.peng_red[i];
+			json_state["penguins"]["blue"][i] = state.peng_blue[i];
+		}
 
 		json_state["score"]["red"] = state.score_red;
 		json_state["score"]["blue"] = state.score_blue;