diff --git a/src/gt4py/next/program_processors/runners/dace_iterator/itir_to_tasklet.py b/src/gt4py/next/program_processors/runners/dace_iterator/itir_to_tasklet.py
index 5b9f48bfcd..053f8b0c7e 100644
--- a/src/gt4py/next/program_processors/runners/dace_iterator/itir_to_tasklet.py
+++ b/src/gt4py/next/program_processors/runners/dace_iterator/itir_to_tasklet.py
@@ -1192,9 +1192,11 @@ def _visit_reduce(self, node: itir.FunCall):
             )
 
             input_mapping = {
-                param: dace.Memlet(data=arg.value.data, subset="0")
-                if arg.value.desc(self.context.body).shape == (1,)
-                else create_memlet_at(arg.value.data, nreduce_index)
+                param: (
+                    dace.Memlet(data=arg.value.data, subset="0")
+                    if arg.value.desc(self.context.body).shape == (1,)
+                    else create_memlet_at(arg.value.data, nreduce_index)
+                )
                 for (param, _), arg in zip(inner_inputs, args)
             }
             output_mapping = {