diff --git a/src/autoplex/auto/phonons/jobs.py b/src/autoplex/auto/phonons/jobs.py
index 82d3a0e0..f8a5a670 100644
--- a/src/autoplex/auto/phonons/jobs.py
+++ b/src/autoplex/auto/phonons/jobs.py
@@ -398,13 +398,15 @@ def complete_benchmark(  # this function was put here to prevent circular import
     if isinstance(dft_references, list):
         return Response(
             replace=Flow(jobs),
-            output={"bm_output": collect_output, "dft_references": dft_references[ibenchmark_structure]},
-        )
-    else:
-        return Response(
-            replace=Flow(jobs),
-            output={"bm_output": collect_output, "dft_references": dft_references},
+            output={
+                "bm_output": collect_output,
+                "dft_references": dft_references[ibenchmark_structure],
+            },
         )
+    return Response(
+        replace=Flow(jobs),
+        output={"bm_output": collect_output, "dft_references": dft_references},
+    )
 
 
 @job