diff --git a/tests/store_model_in_db_tests/test_mcp_servers.py b/tests/store_model_in_db_tests/test_mcp_servers.py index 8e0f11aee1..8197ce2693 100644 --- a/tests/store_model_in_db_tests/test_mcp_servers.py +++ b/tests/store_model_in_db_tests/test_mcp_servers.py @@ -379,7 +379,7 @@ async def test_edit_mcp_server_redacts_credentials(): mock_prisma = mock.Mock() mock_get_prisma.return_value = mock_prisma - mock_manager.add_update_server = mock.Mock() + mock_manager.add_update_server = mock.AsyncMock() mock_manager.reload_servers_from_database = mock.AsyncMock() server_id = str(uuid.uuid4())