diff --git a/olmo/scaling/ladder.py b/olmo/scaling/ladder.py
index 7086be7ab..8a5e01511 100644
--- a/olmo/scaling/ladder.py
+++ b/olmo/scaling/ladder.py
@@ -525,7 +525,7 @@ def train_cmd(args: argparse.Namespace):
     prepare_cli_environment()
     add_cached_path_clients()
 
-    from train import main
+    from olmo.train import main
 
     main(cfg)