diff --git a/tests/test_llm.py b/tests/test_llm.py
index 74453a4b..1dd39dcb 100644
--- a/tests/test_llm.py
+++ b/tests/test_llm.py
@@ -54,6 +54,6 @@ def test_generate_detector(openai_llm_mock):  # noqa: F811
 def test_generate_segmentor(openai_llm_mock):  # noqa: F811
     llm = OpenAILLM()
     prompt = "Can you generate a cat segmentor?"
-    segmentor = llm.generate_detector(prompt)
-    assert isinstance(segmentor, GroundingDINO)
+    segmentor = llm.generate_segmentor(prompt)
+    assert isinstance(segmentor, GroundingSAM)
     assert segmentor.prompt == "cat"