diff --git a/schemas/SpectrumXds.json b/schemas/SpectrumXds.json
index 8de45c91..08ecf420 100644
--- a/schemas/SpectrumXds.json
+++ b/schemas/SpectrumXds.json
@@ -925,7 +925,11 @@
         ]
       ],
       "dtypes": [
-        "|b1"
+        "|b1",
+        "|u1",
+        "<u2",
+        "<u4",
+        "<u8"
       ],
       "coordinates": [
         {
diff --git a/schemas/VisibilityXds.json b/schemas/VisibilityXds.json
index 9f0ebbd1..9c32484b 100644
--- a/schemas/VisibilityXds.json
+++ b/schemas/VisibilityXds.json
@@ -1019,7 +1019,11 @@
         ]
       ],
       "dtypes": [
-        "|b1"
+        "|b1",
+        "|u1",
+        "<u2",
+        "<u4",
+        "<u8"
       ],
       "coordinates": [
         {
diff --git a/src/xradio/measurement_set/schema.py b/src/xradio/measurement_set/schema.py
index 89798188..38ed1ab5 100644
--- a/src/xradio/measurement_set/schema.py
+++ b/src/xradio/measurement_set/schema.py
@@ -786,7 +786,7 @@ class FlagArray:
             tuple[Time, BaselineId, Frequency, Polarization],
             tuple[Time, AntennaName, Frequency, Polarization],  # SD
         ],
-        bool,
+        Union[bool, numpy.uint8, numpy.uint16, numpy.uint32, numpy.uint64],
     ]
     time: Coordof[TimeCoordArray]
     baseline_id: Optional[Coordof[BaselineArray]]  # Only IF