diff --git a/server/train-ai/AITrainer.mjs b/server/train-ai/AITrainer.mjs
index d19f426..27f115e 100644
--- a/server/train-ai/AITrainer.mjs
+++ b/server/train-ai/AITrainer.mjs
@@ -37,7 +37,8 @@ class AITrainer {
 			JSON.stringify({
 				properties: {
 					rssi_min: -150,
-					rssi_max:  0
+					rssi_max:  0,
+					training_mode: "unified"
 				},
 				index: [{
 					id: "Unified AI (not a real gateway)",
@@ -84,7 +85,8 @@ class AITrainer {
 			JSON.stringify({
 				properties: {
 					rssi_min: -150,
-					rssi_max:  0
+					rssi_max:  0,
+					training_mode: "split"
 				},
 				index
 			})