diff --git a/src/app/agents/models.py b/src/app/agents/models.py
index 42f97f6..1bb4142 100644
--- a/src/app/agents/models.py
+++ b/src/app/agents/models.py
@@ -90,7 +90,7 @@ class AppModels(Enum):
 
         raise ValueError(f"Modello non supportato: {self}")
 
-    def get_agent(self, instructions: str, name: str = "", output: BaseModel | None = None, tools: list[Toolkit] = []) -> Agent:
+    def get_agent(self, instructions: str, name: str = "", output_schema: type[BaseModel] | None = None, tools: list[Toolkit] = []) -> Agent:
         """
         Costruisce un agente con il modello e le istruzioni specificate.
         Args:
@@ -106,5 +106,5 @@ class AppModels(Enum):
             retries=2,
             tools=tools,
             delay_between_retries=5, # seconds
-            output_schema=output.__class__ if output else None # se si usa uno schema di output, lo si passa qui
+            output_schema=output_schema
         )
diff --git a/src/app/agents/pipeline.py b/src/app/agents/pipeline.py
index 2123296..a7d1001 100644
--- a/src/app/agents/pipeline.py
+++ b/src/app/agents/pipeline.py
@@ -30,7 +30,7 @@ class Pipeline:
         model = self.available_models[index]
         self.predictor = model.get_agent(
             PREDICTOR_INSTRUCTIONS,
-            output=PredictorOutput, # type: ignore
+            output_schema=PredictorOutput,
         )
 
     def choose_style(self, index: int):