diff --git a/Code/inference.py b/Code/inference.py
index 212d4c5944edbd54daf8f1f12924e42fedd6d56e..2deb942e79d771af84ce76dec32c9488dd81b148 100644
--- a/Code/inference.py
+++ b/Code/inference.py
@@ -47,7 +47,7 @@ def find_target_position(split_sentence, split_target):
 
 
 pos = find_target_position(split_sentence, split_target)
-target_json = json.loads(pos)
+#target_json = json.loads(pos)
 print(f"The target word is {target_word} and at the position {pos}.")
 
 print("Now enter the label: 0 for literal, 1 for non-literal")
@@ -55,7 +55,7 @@ label = int(input())
 
 
 # Convert to data sample for BERT
-data_sample = [{"sentence": sentence, "pos": target_json, "label": label}]
+data_sample = [{"sentence": sentence, "pos": pos, "label": label}]
 print(data_sample)
 
 tokenizer=AutoTokenizer.from_pretrained("bert-base-uncased")