diff --git a/openhexa/sdk/pipelines/parameter.py b/openhexa/sdk/pipelines/parameter.py
index 8fc32936..409f8e57 100644
--- a/openhexa/sdk/pipelines/parameter.py
+++ b/openhexa/sdk/pipelines/parameter.py
@@ -387,16 +387,19 @@ def __init__(
         self,
         code: str,
         *,
-        type: str
-        | int
-        | bool
-        | S3Connection
-        | CustomConnection
-        | DHIS2Connection
-        | IASOConnection
-        | PostgreSQLConnection
-        | GCSConnection
-        | Dataset,
+        type: type[
+            str
+            | int
+            | bool
+            | float
+            | DHIS2Connection
+            | IASOConnection
+            | PostgreSQLConnection
+            | GCSConnection
+            | S3Connection
+            | CustomConnection
+            | Dataset
+        ],
         name: str | None = None,
         choices: typing.Sequence | None = None,
         help: str | None = None,
@@ -562,17 +565,19 @@ def validate_parameters(parameters: list[Parameter]):
 def parameter(
     code: str,
     *,
-    type: str
-    | int
-    | bool
-    | float
-    | DHIS2Connection
-    | IASOConnection
-    | PostgreSQLConnection
-    | GCSConnection
-    | S3Connection
-    | CustomConnection
-    | Dataset,
+    type: type[
+        str
+        | int
+        | bool
+        | float
+        | DHIS2Connection
+        | IASOConnection
+        | PostgreSQLConnection
+        | GCSConnection
+        | S3Connection
+        | CustomConnection
+        | Dataset
+    ],
     name: str | None = None,
     choices: typing.Sequence | None = None,
     help: str | None = None,