diff --git a/exir/serde/serialize.py b/exir/serde/serialize.py
index a2eb2491067..17b37d8317e 100644
--- a/exir/serde/serialize.py
+++ b/exir/serde/serialize.py
@@ -53,7 +53,7 @@ def __init__(
         module_call_graph: List[ep.ModuleCallEntry],
     ) -> None:
         super().__init__(graph_signature, module_call_graph)
-        self.state_dict: Dict[str, torch.Tensor] = {}  # TODO(T157676982)
+        self.state_dict: Dict[str, torch.Tensor] = {}
 
     def serialize_operator(
         self,