diff --git a/spacy_llm/tests/tasks/test_ner.py b/spacy_llm/tests/tasks/test_ner.py
index 97512ca6..3c180048 100644
--- a/spacy_llm/tests/tasks/test_ner.py
+++ b/spacy_llm/tests/tasks/test_ner.py
@@ -994,11 +994,12 @@ def test_add_label():
 
     for label, definition in [
         ("PERSON", "Every person with the name Jack"),
-        ("LOCATION", None),
+        ("LOCATION", "A geographical location, like a country or a city"),
+        ("COMPANY", None),
     ]:
         llm.add_label(label, definition)
     doc = nlp(text)
-    assert len(doc.ents) == 2
+    assert len(doc.ents) > 1
 
 
 @pytest.mark.external