diff --git a/crates/ml/src/training_profile.rs b/crates/ml/src/training_profile.rs
index fecce585a..223bec6e1 100644
--- a/crates/ml/src/training_profile.rs
+++ b/crates/ml/src/training_profile.rs
@@ -1096,7 +1096,7 @@ mod tests {
 
         // [training] values must have been applied
         assert_eq!(hp.epochs, 200);
-        assert_eq!(hp.batch_size, 8192);
+        assert_eq!(hp.batch_size, 16384);
         assert!((hp.learning_rate - 0.0001).abs() < 1e-10);
 
         // OFI MBP-10 data dir should be set in production profile
@@ -1364,7 +1364,7 @@ mod tests {
 
         // [training] section
         assert_eq!(hp.epochs, 200);
-        assert_eq!(hp.batch_size, 8192);
+        assert_eq!(hp.batch_size, 16384);
         assert!((hp.learning_rate - 0.0001).abs() < 1e-10);
         assert!((hp.gamma - 0.99).abs() < 0.001);
         assert!((hp.reward_scale - 1.0).abs() < 0.01);