diff --git a/onnxruntime/python/tools/quantization/onnx_model.py b/onnxruntime/python/tools/quantization/onnx_model.py
index 510774a11b..0cda0a4a59 100644
--- a/onnxruntime/python/tools/quantization/onnx_model.py
+++ b/onnxruntime/python/tools/quantization/onnx_model.py
@@ -272,7 +272,7 @@ class ONNXModel:
         # initialize sorted_nodes
         sorted_nodes = []
         for input in itertools.chain(self.initializer(), self.model.graph.input):
-            if input_name in deps_to_nodes:
+            if input.name in deps_to_nodes:
                 for node_idx in deps_to_nodes[input.name]:
                     deps_count[node_idx] = deps_count[node_idx] - 1
                     if deps_count[node_idx] == 0:
@@ -293,4 +293,5 @@ class ONNXModel:
 
         assert(e == len(self.graph().node)), "Graph is not a DAG"
         self.graph().ClearField('node')
-        self.graph().node.extend(sorted_nodes)
\ No newline at end of file
+        self.graph().node.extend(sorted_nodes)
+