diff --git a/crates/ml/src/trainers/dqn/smoke_tests/helpers.rs b/crates/ml/src/trainers/dqn/smoke_tests/helpers.rs
index dfd342656..04dfdf4be 100644
--- a/crates/ml/src/trainers/dqn/smoke_tests/helpers.rs
+++ b/crates/ml/src/trainers/dqn/smoke_tests/helpers.rs
@@ -100,7 +100,7 @@ pub(super) fn test_data_dir() -> Option<String> {
         .and_then(|p| p.parent()) // repo root
         .unwrap_or(std::path::Path::new("../.."));
     let test_data = workspace.join("test_data");
-    for sub in &["ES.FUT", "ohlcv/ES.FUT"] {
+    for sub in &["futures-baseline/ES.FUT", "ES.FUT", "ohlcv/ES.FUT"] {
         let candidate = test_data.join(sub);
         if candidate.exists() {
             return Some(candidate.to_string_lossy().into_owned());