diff --git a/modules/util/infer_cloud_products.py b/modules/util/infer_cloud_products.py
index 7cbad5afa2a7e7828aeb7590be0810f6d41a3ad3..d34ac002f75b7f207f6bd67da59a579f4c87f2de 100644
--- a/modules/util/infer_cloud_products.py
+++ b/modules/util/infer_cloud_products.py
@@ -44,10 +44,15 @@ def infer_cloud_products(clvrx_path, output_dir, full_disk=True, satellite='GOES
         t0 = time.time()
         if full_disk:
             cld_frac = nn_cld_frac.run_inference_full_disk(pname, None)
-            cld_opd = nn_cld_opd.run_inference_full_disk(pname, None)
+            cld_opd, opd = nn_cld_opd.run_inference_full_disk(pname, None)
         else:
             cld_frac = nn_cld_frac.run_inference(pname, None)
-            cld_opd = nn_cld_opd.run_inference(pname, None)
+            cld_opd, opd = nn_cld_opd.run_inference(pname, None)
+
+        clr_cat = cld_frac == 0
+        cld_cat = cld_frac == 4
+        cld_opd[clr_cat] = opd[clr_cat]
+        cld_opd[cld_cat] = opd[cld_cat]
 
         write_cld_prods_file_nc4(clvrx_str_time, out_file, cld_frac, cld_opd, x_rad, y_rad, None, None,
                                 satellite=satellite, domain=domain, has_time=True)