diff --git a/py/torch_tensorrt/_compile.py b/py/torch_tensorrt/_compile.py
index e59d6a6f3e..89a965fee7 100644
--- a/py/torch_tensorrt/_compile.py
+++ b/py/torch_tensorrt/_compile.py
@@ -24,7 +24,7 @@
     )
 
 if ENABLED_FEATURES.dynamo_frontend:
-    from torch._export import ExportedProgram
+    from torch.export import ExportedProgram
     from torch_tensorrt.dynamo._compiler import compile as dynamo_compile
     from torch_tensorrt.dynamo._compiler import (
         convert_module_to_trt_engine as dynamo_convert_module_to_trt_engine,