diff --git a/test/test_model_modules/test_abstract_model_class.py b/test/test_model_modules/test_abstract_model_class.py
index dfef68d550b07f824ed38e5c7809c00e5386d115..3a610b1b75f97f96722d36623260aad3b19b4393 100644
--- a/test/test_model_modules/test_abstract_model_class.py
+++ b/test/test_model_modules/test_abstract_model_class.py
@@ -1,4 +1,4 @@
-import keras
+import tensorflow.keras as keras
 import pytest
 
 from mlair import AbstractModelClass
diff --git a/test/test_model_modules/test_advanced_paddings.py b/test/test_model_modules/test_advanced_paddings.py
index 8ca81c42c0b807b28c444badba8d92a255341eb4..c1fe3cd46888e1d42476810ccb2707797acde7b2 100644
--- a/test/test_model_modules/test_advanced_paddings.py
+++ b/test/test_model_modules/test_advanced_paddings.py
@@ -1,4 +1,4 @@
-import keras
+import tensorflow.keras as keras
 import pytest
 
 from mlair.model_modules.advanced_paddings import *
diff --git a/test/test_model_modules/test_flatten_tail.py b/test/test_model_modules/test_flatten_tail.py
index 623d51c07f6b27c8d6238d8a5189dea33837115e..09308ad44751d358e1b05fbf27f7fba0f4ce1818 100644
--- a/test/test_model_modules/test_flatten_tail.py
+++ b/test/test_model_modules/test_flatten_tail.py
@@ -1,4 +1,4 @@
-import keras
+import tensorflow.keras as keras
 import pytest
 from mlair.model_modules.flatten import flatten_tail, get_activation
 
diff --git a/test/test_model_modules/test_inception_model.py b/test/test_model_modules/test_inception_model.py
index 2dfc2c9c1c0510355216769b2ab83152a0a02118..0dc5271c11dd0b30faf2315f26900bf64bb65037 100644
--- a/test/test_model_modules/test_inception_model.py
+++ b/test/test_model_modules/test_inception_model.py
@@ -1,4 +1,4 @@
-import keras
+import tensorflow.keras as keras
 import pytest
 
 from mlair.helpers import PyTestRegex
diff --git a/test/test_model_modules/test_keras_extensions.py b/test/test_model_modules/test_keras_extensions.py
index 78559ee0e54c725d242194133549d8b17699b729..6b41f58055f5d2e60ce721b4dd8777ce422f59f2 100644
--- a/test/test_model_modules/test_keras_extensions.py
+++ b/test/test_model_modules/test_keras_extensions.py
@@ -1,6 +1,6 @@
 import os
 
-import keras
+import tensorflow.keras as keras
 import mock
 import pytest
 
diff --git a/test/test_model_modules/test_loss.py b/test/test_model_modules/test_loss.py
index c993830c5290c9beeec392dfd806354ca02eb490..641c9dd6082f7a4fbd60d4dc2e1a73e7841f2098 100644
--- a/test/test_model_modules/test_loss.py
+++ b/test/test_model_modules/test_loss.py
@@ -1,4 +1,4 @@
-import keras
+import tensorflow.keras as keras
 import numpy as np
 
 from mlair.model_modules.loss import l_p_loss, var_loss, custom_loss
diff --git a/test/test_model_modules/test_model_class.py b/test/test_model_modules/test_model_class.py
index b05fd990c79b881124fa86fcccaeb4d9c1976d5b..b3c6f7d9c6ff2d09c3e7de85c8a68ce19b3de557 100644
--- a/test/test_model_modules/test_model_class.py
+++ b/test/test_model_modules/test_model_class.py
@@ -1,4 +1,4 @@
-import keras
+import tensorflow.keras as keras
 import pytest
 
 from mlair.model_modules.model_class import IntelliO3_ts_architecture
diff --git a/test/test_plotting/test_training_monitoring.py b/test/test_plotting/test_training_monitoring.py
index 18009bc19947bd3318c6f1d220d303c1efeec972..654ed71694d8730ee4952ee82260c59c39b14756 100644
--- a/test/test_plotting/test_training_monitoring.py
+++ b/test/test_plotting/test_training_monitoring.py
@@ -1,6 +1,6 @@
 import os
 
-import keras
+import tensorflow.keras as keras
 import pytest
 
 from mlair.model_modules.keras_extensions import LearningRateDecay
diff --git a/test/test_run_modules/test_training.py b/test/test_run_modules/test_training.py
index ed0d8264326f5299403c47deb46859ccde4a85d7..da63012f2507b034c3ae1affddde93a9a8e1c9f8 100644
--- a/test/test_run_modules/test_training.py
+++ b/test/test_run_modules/test_training.py
@@ -4,7 +4,7 @@ import logging
 import os
 import shutil
 
-import keras
+import tensorflow.keras as keras
 import mock
 import pytest
 from keras.callbacks import History