diff --git a/mlair/model_modules/advanced_paddings.py b/mlair/model_modules/advanced_paddings.py
index 2ada9ee2e9d99044b02924e16a6af4ef35d5e8e7..2dfcbcfa8386428b81bb46b08bfbc3c7ae3cf7e5 100644
--- a/mlair/model_modules/advanced_paddings.py
+++ b/mlair/model_modules/advanced_paddings.py
@@ -257,7 +257,7 @@ class _ZeroPadding(Layer):
         self.rank = len(padding)
         self.padding = padding
         self.data_format = normalize_data_format(data_format)
-        self.input_spec = InputSpec(ndim=self.rank + 2)
+        self.input_spec = tf.keras.layers.InputSpec(ndim=self.rank + 2)
         super(_ZeroPadding, self).__init__(**kwargs)
 
     def call(self, inputs):
diff --git a/mlair/model_modules/model_class.py b/mlair/model_modules/model_class.py
index 1165999ebea04618e136268bd7440d4d5989233e..00101566aada90dbb5024a33655048521082df09 100644
--- a/mlair/model_modules/model_class.py
+++ b/mlair/model_modules/model_class.py
@@ -346,7 +346,7 @@ class MyTowerModel(AbstractModelClass):
         self.model = keras.Model(inputs=X_input, outputs=[out_main])
 
     def set_compile_options(self):
-        self.optimizer = keras.optimizers.adam(lr=self.initial_lr)
+        self.optimizer = keras.optimizers.Adam(lr=self.initial_lr)
         self.compile_options = {"loss": [keras.losses.mean_squared_error], "metrics": ["mse"]}
 
 
@@ -457,7 +457,7 @@ class IntelliO3_ts_architecture(AbstractModelClass):
         self.model = keras.Model(inputs=X_input, outputs=[out_minor1, out_main])
 
     def set_compile_options(self):
-        self.compile_options = {"optimizer": keras.optimizers.adam(lr=self.initial_lr, amsgrad=True),
+        self.compile_options = {"optimizer": keras.optimizers.Adam(lr=self.initial_lr, amsgrad=True),
                                 "loss": [l_p_loss(4), keras.losses.mean_squared_error],
                                 "metrics": ['mse'],
                                 "loss_weights": [.01, .99]