diff --git a/model_training_nnn_tpu/rnn_model_tf.py b/model_training_nnn_tpu/rnn_model_tf.py
index 15fff17..14dd242 100644
--- a/model_training_nnn_tpu/rnn_model_tf.py
+++ b/model_training_nnn_tpu/rnn_model_tf.py
@@ -52,13 +52,13 @@ class NoiseModel(keras.Model):
             weight = self.add_weight(
                 name=f'day_weight_{i}',
                 shape=(neural_dim, neural_dim),
-                initializer='identity',
+                initializer=tf.keras.initializers.Identity(),
                 trainable=True
             )
             bias = self.add_weight(
                 name=f'day_bias_{i}',
                 shape=(neural_dim,),
-                initializer='zeros',
+                initializer=tf.keras.initializers.Zeros(),
                 trainable=True
             )
             self.day_weights.append(weight)
@@ -220,13 +220,13 @@ class CleanSpeechModel(keras.Model):
             weight = self.add_weight(
                 name=f'day_weight_{i}',
                 shape=(neural_dim, neural_dim),
-                initializer='identity',
+                initializer=tf.keras.initializers.Identity(),
                 trainable=True
             )
             bias = self.add_weight(
                 name=f'day_bias_{i}',
                 shape=(neural_dim,),
-                initializer='zeros',
+                initializer=tf.keras.initializers.Zeros(),
                 trainable=True
             )
             self.day_weights.append(weight)