diff --git a/modules/deeplearning/cloud_fraction_fcn_viirs.py b/modules/deeplearning/cloud_fraction_fcn_viirs.py
index 3a0d5d55bb9f57424caba804bb4c7cb0c4dae66e..28f76db13c8e70d3f0c191c862ec141a95a1b0e9 100644
--- a/modules/deeplearning/cloud_fraction_fcn_viirs.py
+++ b/modules/deeplearning/cloud_fraction_fcn_viirs.py
@@ -18,7 +18,7 @@ LOG_DEVICE_PLACEMENT = False
 PROC_BATCH_SIZE = 4
 PROC_BATCH_BUFFER_SIZE = 5000
 
-NumClasses = 3
+NumClasses = 5
 if NumClasses == 2:
     NumLogits = 1
 else: