diff --git a/src/finn/transformation/fpgadataflow/templates.py b/src/finn/transformation/fpgadataflow/templates.py
index ebb5ef964ceac4e823e6fba7a6494832954a9358..8749532c38e0951151f30ae9b3875a273bed4c42 100644
--- a/src/finn/transformation/fpgadataflow/templates.py
+++ b/src/finn/transformation/fpgadataflow/templates.py
@@ -103,5 +103,5 @@ dma.recvchannel.wait()
 obuf_folded = packed_bytearray_to_finnpy(obuf_packed, odt, oshape_folded)
 # convert to normal reshape and save
 obuf_normal = obuf_folded.reshape(oshape_normal)
-np.save("output.npy", obuf_unpacked)
+np.save("output.npy", obuf_normal)
 """