diff --git a/test/test_model_modules/test_abstract_model_class.py b/test/test_model_modules/test_abstract_model_class.py
index 2c22b39166d513a28b411ec021cc5af449c7d1e1..06002da1228caf7e28fad6c741e30c2fcd31efd2 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