diff --git a/src/refined/inference/processor.py b/src/refined/inference/processor.py
index 667dc68..b324327 100644
--- a/src/refined/inference/processor.py
+++ b/src/refined/inference/processor.py
@@ -374,7 +374,7 @@ def process_tensors(self, batch: BatchedElementsTns, ner_threshold: float = 0.5,
             span.entity_linking_model_confidence_score = predicted_entity_confidence[span_idx]
             span.top_k_predicted_entities = [
                 (Entity(wikidata_entity_id=f'Q{entity_id}',
-                        wikipedia_entity_title=self.preprocessor.qcode_to_wiki.get(wikidata_id)
+                        wikipedia_entity_title=self.preprocessor.qcode_to_wiki.get(f'Q{entity_id}')
                         if self.preprocessor.qcode_to_wiki is not None else None
                         ),
                  round(score, 4))