diff --git a/saxml/server/servable_model_params.py b/saxml/server/servable_model_params.py
index 9a42ba8..8872afb 100644
--- a/saxml/server/servable_model_params.py
+++ b/saxml/server/servable_model_params.py
@@ -117,6 +117,8 @@ def sax_registration_name(cls) -> Optional[str]:
     """Returns an optional custom registration name for the model."""
     return None
 
+  # TODO(jwtan): Make this a classmethod as most of the config attributes are
+  # class attributes.
   def apply_model_overrides(self, overrides: Dict[str, Any]) -> None:
     """Applies model config overrides received from Publish.