diff --git a/test/test_main.py b/test/test_main.py
index bdfdf76cc5ca77587dcf8c5f298a5acbd10cb29e..f11ee0efcf411aa17d2526f3f9a287f35ea0bed6 100644
--- a/test/test_main.py
+++ b/test/test_main.py
@@ -41,7 +41,7 @@ from declearn.test_utils import run_as_processes
 # pylint: disable=ungrouped-imports
 FRAMEWORKS = ["Sksgd", "Tflow", "Torch"]
 try:
-    import tensorflow as tf
+    import tensorflow as tf  # type: ignore
 except ModuleNotFoundError:
     FRAMEWORKS.remove("Tflow")
 else: