diff --git a/agents/bart_agent.py b/agents/bart_agent.py
index 894f060e3a49c9207dafbbdea801654bdb62680a..36864f501c799866334b2a1bf0ebacd138fc32e2 100644
--- a/agents/bart_agent.py
+++ b/agents/bart_agent.py
@@ -50,7 +50,7 @@ class BARTResponseAgent(object):
         persona = [tokenize(line.strip()) for line in conversation['persona B']]
         # partner = [tokenize(line.strip()) for line in conversation['persona A']]
         partner = [] # Baseline not trained with the partner personaj
-        history = [tokenize(line['text'].strip()) for line in conversation['dialogue'][:-1]]
+        history = [tokenize(line['text'].strip()) for line in conversation['dialogue']]
         return persona, partner, history
    
     def prepare_tensors(self, conversation):