diff --git a/integrations/cohere/tests/test_cohere_chat_generator.py b/integrations/cohere/tests/test_cohere_chat_generator.py
index 9a822856e..1684046eb 100644
--- a/integrations/cohere/tests/test_cohere_chat_generator.py
+++ b/integrations/cohere/tests/test_cohere_chat_generator.py
@@ -308,7 +308,7 @@ def test_live_run_with_connector(self):
         message: ChatMessage = results["replies"][0]
         assert "Paris" in message.content
         assert message.meta["documents"] is not None
-        assert message.meta["citations"] is not None
+        assert "citations" in message.meta  # Citations might be None
 
     @pytest.mark.skipif(
         not os.environ.get("COHERE_API_KEY", None) and not os.environ.get("CO_API_KEY", None),