diff --git a/src/model_modules/model_class.py b/src/model_modules/model_class.py
index 0e31cd66fd64d19f1cecc7c9906a5c3b9446fe75..6b0fe236ff8ee726c34a721a6be0ed8be91f2bb8 100644
--- a/src/model_modules/model_class.py
+++ b/src/model_modules/model_class.py
@@ -219,15 +219,15 @@ class MyBranchedModel(AbstractModelClass):
         x_in = keras.layers.Dense(64, name='{}_Dense_64'.format("major"))(x_in)
         x_in = self.activation()(x_in)
         out_minor_1 = keras.layers.Dense(self.window_lead_time, name='{}_Dense'.format("minor_1"))(x_in)
-        out_minor_1 = self.activation()(out_minor_1)
+        out_minor_1 = self.activation(name="minor_1")(out_minor_1)
         x_in = keras.layers.Dense(32, name='{}_Dense_32'.format("major"))(x_in)
         x_in = self.activation()(x_in)
         out_minor_2 = keras.layers.Dense(self.window_lead_time, name='{}_Dense'.format("minor_2"))(x_in)
-        out_minor_2 = self.activation()(out_minor_2)
+        out_minor_2 = self.activation(name="minor_2")(out_minor_2)
         x_in = keras.layers.Dense(16, name='{}_Dense_16'.format("major"))(x_in)
         x_in = self.activation()(x_in)
         x_in = keras.layers.Dense(self.window_lead_time, name='{}_Dense'.format("major"))(x_in)
-        out_main = self.activation()(x_in)
+        out_main = self.activation(name="main")(x_in)
         self.model = keras.Model(inputs=x_input, outputs=[out_minor_1, out_minor_2, out_main])
 
     def set_loss(self):