diff --git a/luxonis_ml/data/datasets/annotation.py b/luxonis_ml/data/datasets/annotation.py
index e5342cc0..9bca54d8 100644
--- a/luxonis_ml/data/datasets/annotation.py
+++ b/luxonis_ml/data/datasets/annotation.py
@@ -8,7 +8,7 @@
 import pycocotools.mask as mask_util
 from PIL import Image, ImageDraw
 from pydantic import ConfigDict, Field, field_validator, model_validator
-from pydantic.types import FilePath, PositiveInt
+from pydantic.types import FilePath, NonNegativeInt, PositiveInt
 from typing_extensions import Annotated, TypeAlias
 
 from luxonis_ml.utils import BaseModelExtraForbid, Registry
@@ -229,7 +229,7 @@ class RLESegmentationAnnotation(SegmentationAnnotation):
 
     height: PositiveInt
     width: PositiveInt
-    counts: Union[List[PositiveInt], bytes]
+    counts: Union[List[NonNegativeInt], bytes]
 
     def get_value(self) -> Dict[str, Any]:
         if isinstance(self.counts, bytes):
@@ -438,12 +438,12 @@ def to_parquet_dict(self) -> ParquetDict:
             "file": self.file.name,
             "type": self.annotation.__class__.__name__,
             "created_at": datetime.utcnow(),
-            "class": self.annotation.class_ or ""
-            if self.annotation is not None
-            else "",
-            "instance_id": self.annotation.instance_id or -1
-            if self.annotation is not None
-            else -1,
+            "class": (
+                self.annotation.class_ or "" if self.annotation is not None else ""
+            ),
+            "instance_id": (
+                self.annotation.instance_id or -1 if self.annotation is not None else -1
+            ),
             "task": self.annotation.task if self.annotation is not None else "",
             "annotation": json_value,
         }