diff --git a/python/tvm/contrib/pybuda_compile.py b/python/tvm/contrib/pybuda_compile.py
index c0b72047d..260867fa4 100644
--- a/python/tvm/contrib/pybuda_compile.py
+++ b/python/tvm/contrib/pybuda_compile.py
@@ -489,12 +489,13 @@ def duplicate_dequantize_nodes_in_onnx_graph(onnx_module):
             for i, consumer_name in enumerate(consumers):
                 new_node_name = node.name + f"_clone{i}"
                 new_output_name = output_name + f"_clone{i}"
-
+                attrs = {"axis": node.attribute[0].i} if len(node.attribute) > 0 else {}
                 cloned_node = onnx.helper.make_node(
                     node.op_type,
                     node.input,
                     [new_output_name],
-                    name=new_node_name
+                    name=new_node_name,
+                    **attrs
                 )
 
                 # Add the cloned node to the list of nodes to add