diff --git a/inference.py b/inference.py
index 2338d3353a5beaaefe42e153f8f27911a9597cc2..313bfad932dc225bacb1d7e85d7ed6dd425bc283 100644
--- a/inference.py
+++ b/inference.py
@@ -65,7 +65,7 @@ label = int(input())
 
 
 # Convert to data sample for BERT
-data_sample = [{"sentence": sentence, "pos": pos, "label": label}]
+data_sample = [{"sentence": split_sentence, "pos": pos, "label": label}]
 #print(data_sample)
 
 tokenizer = AutoTokenizer.from_pretrained("bert-base-uncased")