diff --git a/model/supervised_finetuning/custom_datasets/toxic_conversation.py b/model/supervised_finetuning/custom_datasets/toxic_conversation.py
index d3433106..640b8d8d 100644
--- a/model/supervised_finetuning/custom_datasets/toxic_conversation.py
+++ b/model/supervised_finetuning/custom_datasets/toxic_conversation.py
@@ -67,4 +67,4 @@ class ProsocialDialogue(Dataset):
         return len(self.pairs)
 
     def __getitem__(self, idx):
-        return format_pair(self.pairs[idx])
+        return self.pairs[idx]