diff --git a/test/models/test_cohere_model.py b/test/models/test_cohere_model.py
index 3174d682f5..8426d733ca 100644
--- a/test/models/test_cohere_model.py
+++ b/test/models/test_cohere_model.py
@@ -178,7 +178,7 @@ def test_cohere_model_run(mock_cohere, mock_record, model_type):
     "model_type",
     model_types,
 )
-@patch("camel.models.cohere_model.record")
+@patch("cohere.ClientV2")
 def test_cohere_model_run_unsupported_role(mock_cohere, model_type):
     # Mock the Cohere create client function
     mock_cohere_client = MagicMock()