diff --git a/apex/contrib/sparsity/permutation_lib.py b/apex/contrib/sparsity/permutation_lib.py
index 375143410..55f6b2029 100644
--- a/apex/contrib/sparsity/permutation_lib.py
+++ b/apex/contrib/sparsity/permutation_lib.py
@@ -2009,13 +2009,7 @@ def trace_and_print_raw_fx_graph(cls, model, print_tabular=False, generate_pytho
             if not torch.distributed.is_initialized() or torch.distributed.get_rank() == 0:
                 if cls.__verbosity > 0:
                     print(ex)
-                    print(
-                        "".join(
-                            traceback.format_exception(
-                                etype=type(ex), value=ex, tb=ex.__traceback__
-                            )
-                        )
-                    )
+                    print("".join(traceback.format_exception(value=ex, tb=ex.__traceback__)))
                     print(
                         "\n[print_raw_fx_graph] Meet the fatal fault when trying to symbolic trace the model with Torch.FX"
                     )