diff --git a/src/maxtext/trainers/post_train/sft/train_sft.py b/src/maxtext/trainers/post_train/sft/train_sft.py
index fb2926e67f..c99b5f48b6 100644
--- a/src/maxtext/trainers/post_train/sft/train_sft.py
+++ b/src/maxtext/trainers/post_train/sft/train_sft.py
@@ -68,6 +68,7 @@
 from maxtext.utils import lora_utils
 from maxtext.utils import max_utils
 from maxtext.utils import max_logging
+# Placeholder: internal
 from maxtext.utils import maxtext_utils
 from maxtext.utils import model_creation_utils