diff --git a/requirements.txt b/requirements.txt
index d44ac3b..481f7e7 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,3 +1,3 @@
 pydantic==2.8.2
 requests
-openai == 1.24.0
+openai == 1.55.3
diff --git a/tests/unit/test_unit.py b/tests/unit/test_unit.py
index daeac79..b7b2c87 100644
--- a/tests/unit/test_unit.py
+++ b/tests/unit/test_unit.py
@@ -53,7 +53,7 @@ def check_ds_created(ds, mock_post, url):
             self._compare_ds(ds, example_ds)
             args, kwargs = mock_post.call_args
 
-            assert kwargs['headers'] == {'Authorization': 'Bearer ' + API_KEY}
+            assert kwargs['headers'] == {'Authorization': 'Bearer ' + API_KEY, 'Content-Type': 'application/json'}
             assert kwargs['json'] == example_ds.model_dump()
             assert args[0] == url
 
@@ -142,14 +142,15 @@ def test_create_knowledge_bases(self, mock_post, mock_get):
 
         args, kwargs = mock_post.call_args
 
-        assert kwargs['headers'] == {'Authorization': 'Bearer ' + API_KEY}
+        assert kwargs['headers'] == {'Authorization': 'Bearer ' + API_KEY, 'Content-Type': 'application/json'}
 
         expected_create_request = {
             'name': test_knowledge_base_config.name,
             'description': test_knowledge_base_config.description,
             'vector_store': {
                 'engine': test_vector_store_config.engine,
-                'connection_data': test_vector_store_config.connection_data
+                'connection_data': test_vector_store_config.connection_data,
+                'table': test_vector_store_config.table
             },
             'embedding_model': {
                 'provider': test_embedding_config.provider,