diff --git a/aimodel/src/lib/ai/RainfallWaterContraster.py b/aimodel/src/lib/ai/RainfallWaterContraster.py
index 1d1998d..3196dc7 100644
--- a/aimodel/src/lib/ai/RainfallWaterContraster.py
+++ b/aimodel/src/lib/ai/RainfallWaterContraster.py
@@ -72,6 +72,7 @@ class RainfallWaterContraster(object):
 		
 		self.model_predict = tf.keras.models.load_model(filepath_checkpoint, custom_objects={
 			"LayerContrastiveEncoder": LayerContrastiveEncoder,
+			"LayerConvNeXtGamma": LayerConvNeXtGamma,
 			"LayerCheeseMultipleOut": LayerCheeseMultipleOut
 		})