diff --git a/case_studies/pos_tagging/scripts/latin_sequence_labeling.py b/case_studies/pos_tagging/scripts/latin_sequence_labeling.py
index b067160..10d1e83 100644
--- a/case_studies/pos_tagging/scripts/latin_sequence_labeling.py
+++ b/case_studies/pos_tagging/scripts/latin_sequence_labeling.py
@@ -111,7 +111,7 @@ def forward(self, input_ids, token_type_ids=None, attention_mask=None, transform
 		if labels is not None:
 			labels = labels.to(device)
 
-		sequence_outputs, pooled_outputs = self.bert(input_ids, token_type_ids=None, attention_mask=attention_mask)
+		sequence_outputs, pooled_outputs = self.bert(input_ids, token_type_ids=None, attention_mask=attention_mask).values()
 		all_layers=sequence_outputs
 		out=torch.matmul(transforms,all_layers)