diff --git a/fx2ait/fx2ait/tools/common_aten2ait.py b/fx2ait/fx2ait/tools/common_aten2ait.py
index 1a603848a..60a4cf4f6 100644
--- a/fx2ait/fx2ait/tools/common_aten2ait.py
+++ b/fx2ait/fx2ait/tools/common_aten2ait.py
@@ -107,7 +107,6 @@ def generate_graph(
                 mod,
                 tuple(original_inputs),
                 CaptureConfig(
-                    pt2_mode=True,
                     enable_functionalization=False,
                     enable_dynamic_shape=True,
                     _use_old_decomp_table=True,