diff --git a/unsloth_zoo/dataset_utils.py b/unsloth_zoo/dataset_utils.py
index f4791bf0..9bb74b9d 100644
--- a/unsloth_zoo/dataset_utils.py
+++ b/unsloth_zoo/dataset_utils.py
@@ -547,7 +547,7 @@ def sft_prepare_dataset(
                 )
             test_text = test_text[0]
         else:
-            test_text = next(iter(dataset))[dataset_text_field][0]
+            test_text = next(iter(dataset))[dataset_text_field]
 
         # Get chat template
         chat_template = getattr(processing_class, 'chat_template', '')