diff --git a/aixplain/factories/model_factory.py b/aixplain/factories/model_factory.py
index a16ea9eb..0fb845f1 100644
--- a/aixplain/factories/model_factory.py
+++ b/aixplain/factories/model_factory.py
@@ -72,7 +72,7 @@ def _create_model_from_response(cls, response: Dict) -> Model:
             supplier=response["supplier"],
             api_key=response["api_key"],
             cost=response["pricing"],
-            function=Function(response["function"]["id"]),
+            function=function,
             parameters=parameters,
             is_subscribed=True if "subscription" in response else False,
             version=response["version"]["id"],
diff --git a/aixplain/modules/model/__init__.py b/aixplain/modules/model/__init__.py
index e631211f..285cbe55 100644
--- a/aixplain/modules/model/__init__.py
+++ b/aixplain/modules/model/__init__.py
@@ -267,7 +267,7 @@ def check_finetune_status(self, after_epoch: Optional[int] = None):
         """
         from aixplain.enums.asset_status import AssetStatus
         from aixplain.modules.finetune.status import FinetuneStatus
-
+        
         headers = {"x-api-key": self.api_key, "Content-Type": "application/json"}
         resp = None
         try: