diff --git a/freqtrade/freqai/prediction_models/PyTorchClassifierMultiTarget.py b/freqtrade/freqai/prediction_models/PyTorchClassifierMultiTarget.py
index 62bec0fd9..f33248e7d 100644
--- a/freqtrade/freqai/prediction_models/PyTorchClassifierMultiTarget.py
+++ b/freqtrade/freqai/prediction_models/PyTorchClassifierMultiTarget.py
@@ -127,7 +127,7 @@ class PyTorchClassifierMultiTarget(BasePyTorchModel):
                 f"expecting labels: {self.multiclass_names}"
             )
 
-    def decode_classes_name(self, classes: torch.Tensor[int]) -> List[str]:
+    def decode_classes_name(self, classes: torch.Tensor) -> List[str]:
         """
         decode class name int -> str
         """