diff --git a/flaml/automl/model.py b/flaml/automl/model.py
index 08d40173a..4df09294d 100644
--- a/flaml/automl/model.py
+++ b/flaml/automl/model.py
@@ -571,7 +571,9 @@ class TransformersEstimator(BaseEstimator):
     @property
     def data_collator(self):
         from flaml.automl.task.task import Task
-        from flaml.automl.nlp.huggingface.data_collator import task_to_datacollator_class
+        from flaml.automl.nlp.huggingface.data_collator import (
+            task_to_datacollator_class,
+        )
 
         data_collator_class = task_to_datacollator_class.get(
             self._task.name if isinstance(self._task, Task) else self._task
@@ -2040,7 +2042,7 @@ class TS_SKLearn(SKLearnEstimator):
         space.update(
             {
                 "optimize_for_horizon": {
-                    "domain": tune.choice([False]),
+                    "domain": tune.choice([True, False]),
                     "init_value": False,
                     "low_cost_init_value": False,
                 },