diff --git a/src/ner/agri_ner_akai/local/model.py b/src/ner/agri_ner_akai/local/model.py
index 4c1cdaa..651c048 100644
--- a/src/ner/agri_ner_akai/local/model.py
+++ b/src/ner/agri_ner_akai/local/model.py
@@ -45,9 +45,10 @@ def aggregate_entities(sentence, entity_outputs):
 
                 else:
                     # If it's a subword but not an I- entity
-                    current_entity["word"] += entity["word"][2:]
-                    current_entity["end"] = entity["end"]
-                    current_entity["score"] = float((current_entity["score"] + entity["score"]) / 2)  # averaging scores
+                    if current_entity:
+                        current_entity["word"] += entity["word"][2:]
+                        current_entity["end"] = entity["end"]
+                        current_entity["score"] = float((current_entity["score"] + entity["score"]) / 2)  # averaging scores
 
             # Handle full words
             else: