diff --git a/analysis/compare_marin_to_olmo.py b/analysis/compare_marin_to_olmo.py
index 051daa15f..53eae94a2 100644
--- a/analysis/compare_marin_to_olmo.py
+++ b/analysis/compare_marin_to_olmo.py
@@ -125,5 +125,5 @@ def compare_marin_to_olmo_train_loss(marin_run_id: str = "eo302w0523", smooth: b
 
 
 if __name__ == "__main__":
-    # fire.Fire(compare_marin_to_olmo)
-    fire.Fire(compare_marin_to_olmo_train_loss)
+    fire.Fire(compare_marin_to_olmo)
+    # fire.Fire(compare_marin_to_olmo_train_loss)