diff --git a/src/finn/custom_op/fpgadataflow/channelwise_op_batch.py b/src/finn/custom_op/fpgadataflow/channelwise_op_batch.py
index 302a729a9bde357b6bbcbc49695546d49e91c1d9..f6c5624543e47f488e5c42983a324ff48b43decd 100644
--- a/src/finn/custom_op/fpgadataflow/channelwise_op_batch.py
+++ b/src/finn/custom_op/fpgadataflow/channelwise_op_batch.py
@@ -354,9 +354,9 @@ class ChannelwiseOp_Batch(HLSCustomOp):
         elif func == "cmp_ge":
             func_str = "comp::greater_equal<%s, %s>" % (idt_hls, pdt_hls)
         elif func == "add":
-            func_str = "comp::add<%s, %s, %s>" % (idt_hls, pdt_hls, odt_hls)
+            func_str = "comp::add<%s, %s, %s>" % (odt_hls, odt_hls, odt_hls)
         elif func == "mul":
-            func_str = "comp::mul<%s, %s, %s>" % (idt_hls, pdt_hls, odt_hls)
+            func_str = "comp::mul<%s, %s, %s>" % (odt_hls, odt_hls, odt_hls)
         else:
             raise Exception(
                 """Invalid value for attribute Func! Is currently set to: {}