diff --git a/tests/decorators/test_openai.py b/tests/decorators/test_openai.py
index 410b13a9..5d0e4c1c 100644
--- a/tests/decorators/test_openai.py
+++ b/tests/decorators/test_openai.py
@@ -68,8 +68,8 @@ def test_assistant_query_input_valid_creation(self):
         self.assertEqual(input.get_dict_repr(),
                          {"name": "test",
                           "timestampUtc": "timestamp_utc",
-                          "chat_storage_connection_setting": "AzureWebJobsStorage",
-                          "collection_name": "SampleChatState",
+                          "chatstorageconnectionsetting": "AzureWebJobsStorage",
+                          "collectionname": "SampleChatState",
                           "dataType": DataType.UNDEFINED,
                           "direction": BindingDirection.IN,
                           "type": "assistantQuery",
@@ -126,8 +126,8 @@ def test_assistant_post_input_valid_creation(self):
                          {"name": "test",
                           "id": "test_id",
                           "model": "test_model",
-                          "chat_storage_connection_setting": "AzureWebJobsStorage",
-                          "collection_name": "SampleChatState",
+                          "chatstorageconnectionsetting": "AzureWebJobsStorage",
+                          "collectionname": "SampleChatState",
                           "userMessage": "test_message",
                           "dataType": DataType.UNDEFINED,
                           "direction": BindingDirection.IN,