diff --git a/aimodel/src/lib/ai/RainfallWaterMono.py b/aimodel/src/lib/ai/RainfallWaterMono.py
index dbfed1c..873b3d6 100644
--- a/aimodel/src/lib/ai/RainfallWaterMono.py
+++ b/aimodel/src/lib/ai/RainfallWaterMono.py
@@ -15,6 +15,7 @@ from .helpers import make_callbacks
 from .helpers import summarywriter
 from .components.LayerConvNeXtGamma import LayerConvNeXtGamma
 from .components.LayerStack2Image import LayerStack2Image
+from .components.LossCrossentropy import LossCrossentropy
 from .helpers.summarywriter import summarywriter
 
 class RainfallWaterMono(object):
@@ -71,7 +72,8 @@ class RainfallWaterMono(object):
 		
 		self.model = tf.keras.models.load_model(filepath_checkpoint, custom_objects={
 			"LayerConvNeXtGamma": LayerConvNeXtGamma,
-			"LayerStack2Image": LayerStack2Image
+			"LayerStack2Image": LayerStack2Image,
+			"LossCrossentropy": LossCrossentropy
 		})