diff --git a/mlair/model_modules/fully_connected_networks.py b/mlair/model_modules/fully_connected_networks.py
index 45b8eb63ca7afc6e1de5cd3767bd3a11280d3b43..9e3657c36462255cacbb12b2720eb8243e37bc92 100644
--- a/mlair/model_modules/fully_connected_networks.py
+++ b/mlair/model_modules/fully_connected_networks.py
@@ -95,7 +95,7 @@ class FCN(AbstractModelClass):
         # apply to model
         self.set_model()
         self.set_compile_options()
-        # self.set_custom_objects(loss=self.compile_options['loss'])
+        self.set_custom_objects(loss=custom_loss([keras.losses.mean_squared_error, var_loss]), var_loss=var_loss)
 
     def _set_activation(self, activation):
         try: