diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index 9278681..500a31b 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -15,7 +15,7 @@ jobs:
 
     strategy:
       matrix:
-        python-version: [3.7, 3.9, 3.12]
+        python-version: [3.7, 3.9, 3.11]
 
     steps:
       - name: Checkout code
@@ -34,4 +34,4 @@ jobs:
 
       - name: Run lint check
         run: |
-          black --check --verbose deepinfra
\ No newline at end of file
+          black --check --verbose  deepinfra
\ No newline at end of file
diff --git a/deepinfra/models/base/automatic_speech_recognition.py b/deepinfra/models/base/automatic_speech_recognition.py
index 465456f..b8f46d9 100644
--- a/deepinfra/models/base/automatic_speech_recognition.py
+++ b/deepinfra/models/base/automatic_speech_recognition.py
@@ -1,7 +1,4 @@
 from deepinfra.models.base import BaseModel
-from deepinfra.types.automatic_speech_recognition.request import (
-    AutomaticSpeechRecognitionRequest,
-)
 from deepinfra.types.automatic_speech_recognition.response import (
     AutomaticSpeechRecognitionResponse,
 )
@@ -20,9 +17,7 @@ class AutomaticSpeechRecognition(BaseModel):
     def __init__(self, endpoint: str, auth_token: str = None):
         super().__init__(endpoint, auth_token)
 
-    def generate(
-        self, body: AutomaticSpeechRecognitionRequest
-    ) -> AutomaticSpeechRecognitionResponse:
+    def generate(self, body) -> AutomaticSpeechRecognitionResponse:
         """
         Generates the automatic speech recognition response.
         @param body: The request body.
diff --git a/deepinfra/models/base/image_generation.py b/deepinfra/models/base/image_generation.py
index c178a07..1e90ae0 100644
--- a/deepinfra/models/base/image_generation.py
+++ b/deepinfra/models/base/image_generation.py
@@ -1,4 +1,4 @@
-from deepinfra.models.base.base import BaseModel
+from .base import BaseModel
 
 
 class ImageGeneration(BaseModel):