diff --git a/src/rad/resources/schemas/wfi_mosaic-1.0.0.yaml b/src/rad/resources/schemas/wfi_mosaic-1.0.0.yaml
index 130d58eb..47b8f738 100644
--- a/src/rad/resources/schemas/wfi_mosaic-1.0.0.yaml
+++ b/src/rad/resources/schemas/wfi_mosaic-1.0.0.yaml
@@ -37,7 +37,7 @@ properties:
         datatype: float32
         ndim: 2
       unit:
-        tag: tag:astropy.org:astropy/units/unit-1.0.0
+        tag: tag:stsci.edu:asdf/unit/unit-1.0.0
         enum: ["MJy.sr**-1"]
   err:
     tag: tag:stsci.edu:asdf/unit/quantity-1.1.0
@@ -47,7 +47,7 @@ properties:
         datatype: float32
         ndim: 2
       unit:
-        tag: tag:astropy.org:astropy/units/unit-1.0.0
+        tag: tag:stsci.edu:asdf/unit/unit-1.0.0
         enum: ["MJy.sr**-1"]
   context:
     tag: tag:stsci.edu:asdf/core/ndarray-1.0.0
@@ -65,8 +65,8 @@ properties:
         datatype: float32
         ndim: 2
       unit:
-        tag: tag:astropy.org:astropy/units/unit-1.0.0
-        enum: ["(MJy.sr**-1)**2"]
+        tag: tag:stsci.edu:asdf/unit/unit-1.0.0
+        enum: ["MJy**2.sr**-2"]
   var_rnoise:
     tag: tag:stsci.edu:asdf/unit/quantity-1.1.0
     properties:
@@ -75,8 +75,8 @@ properties:
         datatype: float32
         ndim: 2
       unit:
-        tag: tag:astropy.org:astropy/units/unit-1.0.0
-        enum: ["(MJy.sr**-1)**2"]
+        tag: tag:stsci.edu:asdf/unit/unit-1.0.0
+        enum: ["MJy**2.sr**-2"]
   var_flat:
     tag: tag:stsci.edu:asdf/unit/quantity-1.1.0
     properties:
@@ -85,8 +85,8 @@ properties:
         datatype: float32
         ndim: 2
       unit:
-        tag: tag:astropy.org:astropy/units/unit-1.0.0
-        enum: ["(MJy.sr**-1)**2"]
+        tag: tag:stsci.edu:asdf/unit/unit-1.0.0
+        enum: ["MJy**2.sr**-2"]
   cal_logs:
     tag: asdf://stsci.edu/datamodels/roman/tags/cal_logs-1.0.0
 propertyOrder: [meta, data, context, err, weight, var_poisson, var_rnoise, var_flat,