diff --git a/tt_torch/dynamo/backend.py b/tt_torch/dynamo/backend.py
index 23f9c6e..603361f 100644
--- a/tt_torch/dynamo/backend.py
+++ b/tt_torch/dynamo/backend.py
@@ -310,7 +310,7 @@ def run_gm_op_by_op(self, *inputs):
                 node_to_tensor[node] = tensor
                 args_set = set()
                 for arg in node.args:
-                    if arg in args_set: # skip duplicate args
+                    if arg in args_set:  # skip duplicate args
                         continue
                     args_set.add(arg)
                     if isinstance(arg, torch.fx.node.Node):
@@ -319,7 +319,7 @@ def run_gm_op_by_op(self, *inputs):
                 args = node.args[0]
                 output_tensors = [node_to_tensor[arg] for arg in args]
                 outputs = output_tensors
-                
+
                 args_set = set()
                 for arg in args:
                     if arg in args_set: