diff --git a/qurator/sbb_ner/webapp/app.py b/qurator/sbb_ner/webapp/app.py
index 9c92e17..4a23ca0 100644
--- a/qurator/sbb_ner/webapp/app.py
+++ b/qurator/sbb_ner/webapp/app.py
@@ -236,6 +236,9 @@ def ner(model_id=None):
 
                 word += original_text[orig_pos]
 
+                if word_pred != 'X':
+                    last_prediction = word_pred
+
                 continue
 
             if not token.startswith('##') and word_pred == 'X':