diff --git a/server/models/request.py b/server/models/request.py
index efc7024..71461a4 100644
--- a/server/models/request.py
+++ b/server/models/request.py
@@ -10,7 +10,7 @@ class Message(BaseModel): # pylint: disable=too-few-public-methods
     messages: Optional[Any] = None
     temperature: Optional[float or int] = None
 
-    class Config: # pylint: disable=too-few-public-methods
+    class ConfigDict: # pylint: disable=too-few-public-methods
         """A message to send to the chatbot."""
         json_schema_extra = {
             "example": {
diff --git a/server/models/response.py b/server/models/response.py
index 130e859..8f5bc77 100644
--- a/server/models/response.py
+++ b/server/models/response.py
@@ -3,7 +3,7 @@
 
 
 class ResponseStatus(BaseModel):
-    version: str = Field(example='v0.0.15')
+    version: str = Field(default='v0.0.15')
     
 #################################################
 ## ChatGPT
@@ -12,7 +12,7 @@ class ResponseStatus(BaseModel):
 class ResponseChat(BaseModel):
     # data: ChatCompletion
     
-	class Config: # pylint: disable=too-few-public-methods
+	class ConfigDict: # pylint: disable=too-few-public-methods
 		"""A message to send to the chatbot."""
 		json_schema_extra = {
 			"example": {
diff --git a/tests/integration/test_main.py b/tests/integration/test_main.py
index a4c8b08..7b429cf 100644
--- a/tests/integration/test_main.py
+++ b/tests/integration/test_main.py
@@ -1,6 +1,6 @@
 from starlette.testclient import TestClient
 
-from main import app
+from server.api import app
 
 client = TestClient(app)
 
diff --git a/tests/unit/test_example.py b/tests/unit/test_example.py
index 19a64e9..7f8295c 100644
--- a/tests/unit/test_example.py
+++ b/tests/unit/test_example.py
@@ -4,7 +4,7 @@
 
 class TestExampleCase(unittest.TestCase):
     
-    @unittest.skip("Example Test Case")
+    # @unittest.skip("Example Test Case")
     def test_retrieve_files(self):
         """Test that the files are retrieved."""
         token = "test"