diff --git a/deepsensor/model/model.py b/deepsensor/model/model.py
index d32d0f07..cb758462 100644
--- a/deepsensor/model/model.py
+++ b/deepsensor/model/model.py
@@ -243,7 +243,7 @@ def predict(
         aux_at_targets_override: Union[xr.Dataset, xr.DataArray] = None,
         aux_at_targets_override_is_normalised: bool = False,
         resolution_factor: int = 1,
-        pred_params: tuple[str] = ("mean", "std"),
+        pred_params: Tuple[str] = ("mean", "std"),
         n_samples: int = 0,
         ar_sample: bool = False,
         ar_subsample_factor: int = 1,
@@ -274,7 +274,7 @@ def predict(
                 Whether the `aux_at_targets_override` coords are normalised.
                 If False, the DataProcessor will normalise the coords before passing to model.
                 Default False.
-            pred_params (tuple[str]):
+            pred_params (Tuple[str]):
                 Tuple of prediction parameters to return. The strings refer to methods
                 of the model class which will be called and stored in the Prediction object.
                 Default ("mean", "std").