diff --git a/data/text_sampling/text_sampling.py b/data/text_sampling/text_sampling.py
index 5c4e3bbbf..fc2ec5dac 100644
--- a/data/text_sampling/text_sampling.py
+++ b/data/text_sampling/text_sampling.py
@@ -929,7 +929,7 @@ def export(self, dir_suffix: str = None):
                     lm_eval_yaml["training_split"] = "train"
                 if split_out == "valid":
                     lm_eval_yaml["validation_split"] = "validation"
-                if split_out == "valid":
+                if split_out == "test":
                     lm_eval_yaml["test_split"] = "test"
 
             fn_lm_eval_yaml = output_path_dir + "/config.yaml"