diff --git a/source/model/base/BaseModel.py b/source/model/base/BaseModel.py
index ce763b0..d94dfca 100644
--- a/source/model/base/BaseModel.py
+++ b/source/model/base/BaseModel.py
@@ -3,7 +3,6 @@ import gc
 import typing
 from pathlib import Path
 
-from source import api
 from source.manager import ModelManager
 
 
@@ -42,7 +41,7 @@ class BaseModel(abc.ABC):
 
         return {
             "name": self.name,
-            "response_mimetype": self.response_mimetype,
+            "output_type": self.output_type,
         }
 
     def load(self) -> None: