diff --git a/flask_predict_no_batch_t5.py b/flask_predict_no_batch_t5.py
index 860318c..e196151 100644
--- a/flask_predict_no_batch_t5.py
+++ b/flask_predict_no_batch_t5.py
@@ -206,7 +206,7 @@ def predict_data_post_processing(text_list):
 
     return_list = {}
     sentence_one = []
-    sentence_id = "0"
+    sentence_id = text_list_sentence[0][1]
     for i in text_list_sentence:
         if i[1] == sentence_id:
             sentence_one.append(i[0])