diff --git a/exir/serde/export_serialize.py b/exir/serde/export_serialize.py
index 87691dfbee2..13590b78dde 100644
--- a/exir/serde/export_serialize.py
+++ b/exir/serde/export_serialize.py
@@ -22,6 +22,7 @@
 import sympy
 
 import torch
+import torch._export.exported_program
 import torch.export.exported_program as ep
 
 from torch._export.serde.schema import (