diff --git a/src/trainable_entity_extractor/extractors/text_to_multi_option_extractor/TextToMultiOptionMethod.py b/src/trainable_entity_extractor/extractors/text_to_multi_option_extractor/TextToMultiOptionMethod.py
index 017bf9e..32455e8 100644
--- a/src/trainable_entity_extractor/extractors/text_to_multi_option_extractor/TextToMultiOptionMethod.py
+++ b/src/trainable_entity_extractor/extractors/text_to_multi_option_extractor/TextToMultiOptionMethod.py
@@ -57,9 +57,7 @@ def performance(self, extraction_data: ExtractionData) -> float:
 
         self.train(performance_train_set)
 
-        prediction_samples = [
-            PredictionSample(segment_selector_texts=x.segment_selector_texts) for x in performance_test_set.samples
-        ]
+        prediction_samples = [PredictionSample(source_text=x.labeled_data.source_text) for x in performance_test_set.samples]
         predictions = self.predict(prediction_samples)
 
         self.remove_model()