diff --git a/camel_tools/disambig/bert/unfactored.py b/camel_tools/disambig/bert/unfactored.py
index 7089deb..7abd2ab 100644
--- a/camel_tools/disambig/bert/unfactored.py
+++ b/camel_tools/disambig/bert/unfactored.py
@@ -470,7 +470,7 @@ def _scored_analyses(self, word_dd, prediction):
             # return the predictions from BERT
             return [ScoredAnalysis(0,                      # score
                                    bert_analysis,          # analysis
-                                   bert_analysis['diac'],  # diac
+                                   word_dd,                # diac
                                    -99,                    # pos_lex_logprob
                                    -99,                    # lex_logprob
                                    )]