diff --git a/modules/lang/sml/config.el b/modules/lang/sml/config.el
index 382973d4c..0500686dc 100644
--- a/modules/lang/sml/config.el
+++ b/modules/lang/sml/config.el
@@ -41,5 +41,6 @@
 (use-package! company-mlton
   :when (modulep! :completion company)
   :hook (sml-mode . company-mlton-init)
+  :hook (sml-ts-mode . company-mlton-init)
   :config
-  (set-company-backend! 'sml-mode company-mlton-grouped-backend))
+  (set-company-backend! '(sml-mode sml-ts-mode) company-mlton-grouped-backend))