diff --git a/crates/ml/src/cuda_pipeline/gpu_backtest_evaluator.rs b/crates/ml/src/cuda_pipeline/gpu_backtest_evaluator.rs
index 78ba44f8d..e82ceb23e 100644
--- a/crates/ml/src/cuda_pipeline/gpu_backtest_evaluator.rs
+++ b/crates/ml/src/cuda_pipeline/gpu_backtest_evaluator.rs
@@ -2078,8 +2078,8 @@ mod tests {
         assert_eq!(sizes[4], 128 * 256);
         // w_v2: 51 * 128 = 6528
         assert_eq!(sizes[6], 51 * 128);
-        // w_b0out: 9 * 51 * 128 = 58752 (9 exposure actions)
-        assert_eq!(sizes[10], 9 * 51 * 128);
+        // w_b0out: 3 * 51 * 128 = 19584 (3 direction actions — was 9 exposure)
+        assert_eq!(sizes[10], 3 * 51 * 128);
         // w_b1out: 3 * 51 * 128 = 19584
         assert_eq!(sizes[14], 3 * 51 * 128);
         // w_b2out: 3 * 51 * 128 = 19584