diff --git a/crates/ml/tests/smoke_test_real_data.rs b/crates/ml/tests/smoke_test_real_data.rs
index f291c66d9..706b9133f 100644
--- a/crates/ml/tests/smoke_test_real_data.rs
+++ b/crates/ml/tests/smoke_test_real_data.rs
@@ -543,6 +543,7 @@ mod gpu_smoke {
             (64, 64, 32, 32),
             (54, 51, 51),
             4, 50,
+            0.0,   // curiosity_weight (disabled for test)
         ).unwrap();
 
         // Run 4 episodes of 50 timesteps each
@@ -619,6 +620,7 @@ mod gpu_smoke {
             (64, 64, 32, 32),
             (54, 51, 51),
             4, 50,
+            0.0,   // curiosity_weight (disabled for test)
         ).unwrap();
 
         let n_episodes = 4_usize;
@@ -681,6 +683,7 @@ mod gpu_smoke {
             (64, 64, 32, 32),
             (54, 51, 51),
             4, 50,
+            0.0,   // curiosity_weight (disabled for test)
         ).unwrap();
 
         let n_episodes = 4_usize;