diff --git a/orttraining/orttraining/python/training/torchdynamo/register_backend.py b/orttraining/orttraining/python/training/torchdynamo/register_backend.py
index a9450e119c..2830a10b4f 100644
--- a/orttraining/orttraining/python/training/torchdynamo/register_backend.py
+++ b/orttraining/orttraining/python/training/torchdynamo/register_backend.py
@@ -4,7 +4,7 @@
 # --------------------------------------------------------------------------
 
 from functorch.compile import min_cut_rematerialization_partition
-from torch._dynamo.optimizations.training import aot_autograd
+from torch._dynamo.backends.common import aot_autograd
 from .ort_backend import OrtBackend