diff --git a/depthai_nodes/ml/messages/classification.py b/depthai_nodes/ml/messages/classification.py
index 7f539686..911889a1 100644
--- a/depthai_nodes/ml/messages/classification.py
+++ b/depthai_nodes/ml/messages/classification.py
@@ -112,8 +112,10 @@ def transformation(self, value: dai.ImgTransformation):
         @type value: dai.ImgTransformation
         @raise TypeError: If value is not a dai.ImgTransformation object.
         """
-        if not isinstance(value, dai.ImgTransformation):
-            raise TypeError(
-                f"Transformation must be a dai.ImgTransformation object, instead got {type(value)}."
-            )
+
+        if value is not None:
+            if not isinstance(value, dai.ImgTransformation):
+                raise TypeError(
+                    f"Transformation must be a dai.ImgTransformation object, instead got {type(value)}."
+                )
         self._transformation = value
diff --git a/depthai_nodes/ml/messages/clusters.py b/depthai_nodes/ml/messages/clusters.py
index 47095e9b..145d5abb 100644
--- a/depthai_nodes/ml/messages/clusters.py
+++ b/depthai_nodes/ml/messages/clusters.py
@@ -124,8 +124,10 @@ def transformation(self, value: dai.ImgTransformation):
         @type value: dai.ImgTransformation
         @raise TypeError: If value is not a dai.ImgTransformation object.
         """
-        if not isinstance(value, dai.ImgTransformation):
-            raise TypeError(
-                f"Transformation must be a dai.ImgTransformation object, instead got {type(value)}."
-            )
+
+        if value is not None:
+            if not isinstance(value, dai.ImgTransformation):
+                raise TypeError(
+                    f"Transformation must be a dai.ImgTransformation object, instead got {type(value)}."
+                )
         self._transformation = value
diff --git a/depthai_nodes/ml/messages/img_detections.py b/depthai_nodes/ml/messages/img_detections.py
index 71903dc9..b0443698 100644
--- a/depthai_nodes/ml/messages/img_detections.py
+++ b/depthai_nodes/ml/messages/img_detections.py
@@ -247,8 +247,9 @@ def transformation(self, value: dai.ImgTransformation):
         @type value: dai.ImgTransformation
         @raise TypeError: If value is not a dai.ImgTransformation object.
         """
-        if not isinstance(value, dai.ImgTransformation):
-            raise TypeError(
-                f"Transformation must be a dai.ImgTransformation object, instead got {type(value)}."
-            )
+        if value is not None:
+            if not isinstance(value, dai.ImgTransformation):
+                raise TypeError(
+                    f"Transformation must be a dai.ImgTransformation object, instead got {type(value)}."
+                )
         self._transformation = value
diff --git a/depthai_nodes/ml/messages/keypoints.py b/depthai_nodes/ml/messages/keypoints.py
index a65df5bb..27c32d22 100644
--- a/depthai_nodes/ml/messages/keypoints.py
+++ b/depthai_nodes/ml/messages/keypoints.py
@@ -178,8 +178,10 @@ def transformation(self, value: dai.ImgTransformation):
         @type value: dai.ImgTransformation
         @raise TypeError: If value is not a dai.ImgTransformation object.
         """
-        if not isinstance(value, dai.ImgTransformation):
-            raise TypeError(
-                f"Transformation must be a dai.ImgTransformation object, instead got {type(value)}."
-            )
+
+        if value is not None:
+            if not isinstance(value, dai.ImgTransformation):
+                raise TypeError(
+                    f"Transformation must be a dai.ImgTransformation object, instead got {type(value)}."
+                )
         self._transformation = value
diff --git a/depthai_nodes/ml/messages/lines.py b/depthai_nodes/ml/messages/lines.py
index 24d47bda..f9d7015a 100644
--- a/depthai_nodes/ml/messages/lines.py
+++ b/depthai_nodes/ml/messages/lines.py
@@ -151,8 +151,10 @@ def transformation(self, value: dai.ImgTransformation):
         @type value: dai.ImgTransformation
         @raise TypeError: If value is not a dai.ImgTransformation object.
         """
-        if not isinstance(value, dai.ImgTransformation):
-            raise TypeError(
-                f"Transformation must be a dai.ImgTransformation object, instead got {type(value)}."
-            )
+
+        if value is not None:
+            if not isinstance(value, dai.ImgTransformation):
+                raise TypeError(
+                    f"Transformation must be a dai.ImgTransformation object, instead got {type(value)}."
+                )
         self._transformation = value
diff --git a/depthai_nodes/ml/messages/map.py b/depthai_nodes/ml/messages/map.py
index 15ba0064..0158961e 100644
--- a/depthai_nodes/ml/messages/map.py
+++ b/depthai_nodes/ml/messages/map.py
@@ -92,8 +92,10 @@ def transformation(self, value: dai.ImgTransformation):
         @type value: dai.ImgTransformation
         @raise TypeError: If value is not a dai.ImgTransformation object.
         """
-        if not isinstance(value, dai.ImgTransformation):
-            raise TypeError(
-                f"Transformation must be a dai.ImgTransformation object, instead got {type(value)}."
-            )
+
+        if value is not None:
+            if not isinstance(value, dai.ImgTransformation):
+                raise TypeError(
+                    f"Transformation must be a dai.ImgTransformation object, instead got {type(value)}."
+                )
         self._transformation = value
diff --git a/depthai_nodes/ml/messages/prediction.py b/depthai_nodes/ml/messages/prediction.py
index 3d53ada5..29e0cdb6 100644
--- a/depthai_nodes/ml/messages/prediction.py
+++ b/depthai_nodes/ml/messages/prediction.py
@@ -110,8 +110,10 @@ def transformation(self, value: dai.ImgTransformation):
         @type value: dai.ImgTransformation
         @raise TypeError: If value is not a dai.ImgTransformation object.
         """
-        if not isinstance(value, dai.ImgTransformation):
-            raise TypeError(
-                f"Transformation must be a dai.ImgTransformation object, instead got {type(value)}."
-            )
+
+        if value is not None:
+            if not isinstance(value, dai.ImgTransformation):
+                raise TypeError(
+                    f"Transformation must be a dai.ImgTransformation object, instead got {type(value)}."
+                )
         self._transformation = value
diff --git a/depthai_nodes/ml/messages/segmentation.py b/depthai_nodes/ml/messages/segmentation.py
index e2ce9a47..27952281 100644
--- a/depthai_nodes/ml/messages/segmentation.py
+++ b/depthai_nodes/ml/messages/segmentation.py
@@ -69,8 +69,10 @@ def transformation(self, value: dai.ImgTransformation):
         @type value: dai.ImgTransformation
         @raise TypeError: If value is not a dai.ImgTransformation object.
         """
-        if not isinstance(value, dai.ImgTransformation):
-            raise TypeError(
-                f"Transformation must be a dai.ImgTransformation object, instead got {type(value)}."
-            )
+
+        if value is not None:
+            if not isinstance(value, dai.ImgTransformation):
+                raise TypeError(
+                    f"Transformation must be a dai.ImgTransformation object, instead got {type(value)}."
+                )
         self._transformation = value