diff --git a/janus_core/cli/preprocess.py b/janus_core/cli/preprocess.py
index 2eac487d..8045170c 100644
--- a/janus_core/cli/preprocess.py
+++ b/janus_core/cli/preprocess.py
@@ -51,13 +51,17 @@ def preprocess(
 
     inputs = {"mlip_config": str(mlip_config)}
 
-    # Save summary information before preprocessing begins
-    start_summary(command="preprocess", summary=summary, inputs=inputs)
-
     log_kwargs = {"filemode": "w"}
     if log:
         log_kwargs["filename"] = log
 
+    output_files = {"log": log}
+
+    # Save summary information before preprocessing begins
+    start_summary(
+        command="preprocess", summary=summary, inputs=inputs, output_files=output_files
+    )
+
     # Run preprocessing
     run_preprocess(
         mlip_config, attach_logger=True, log_kwargs=log_kwargs, track_carbon=tracker
diff --git a/janus_core/cli/train.py b/janus_core/cli/train.py
index 52077dfa..0ffd28a0 100644
--- a/janus_core/cli/train.py
+++ b/janus_core/cli/train.py
@@ -77,13 +77,17 @@ def train(
 
     inputs = {"mlip_config": str(mlip_config), "fine_tune": fine_tune}
 
-    # Save summary information before training begins
-    start_summary(command="train", summary=summary, inputs=inputs)
-
     log_kwargs = {"filemode": "w"}
     if log:
         log_kwargs["filename"] = log
 
+    output_files = {"log": log}
+
+    # Save summary information before training begins
+    start_summary(
+        command="train", summary=summary, inputs=inputs, output_files=output_files
+    )
+
     # Run training
     run_train(
         mlip_config, attach_logger=True, log_kwargs=log_kwargs, track_carbon=tracker