diff --git a/VAE.Keras3/modules/layers/VariationalLossLayer.py b/VAE.Keras3/modules/layers/VariationalLossLayer.py
index 8e9610c56187a4d768f069d392a23f4ac21c2ac8..db15b4cf93a9394a691fc4d1d34af24d74bcd85c 100644
--- a/VAE.Keras3/modules/layers/VariationalLossLayer.py
+++ b/VAE.Keras3/modules/layers/VariationalLossLayer.py
@@ -25,7 +25,8 @@ class VariationalLossLayer(keras.layers.Layer):
 
     def call(self, inputs):
         
-        k1,k2 = self.loss_weights
+        k1 = self.k1
+        k2 = self.k2
         
         # ---- Retrieve inputs
         #