diff --git a/openfl/federated/task/__init__.py b/openfl/federated/task/__init__.py
index 084fc9506c..0838fc802d 100644
--- a/openfl/federated/task/__init__.py
+++ b/openfl/federated/task/__init__.py
@@ -16,8 +16,7 @@
 from .runner import TaskRunner  # NOQA
 
 if importlib.util.find_spec('tensorflow'):
-    from .runner_tf import TensorFlowTaskRunner  # NOQA
-    from .runner_tf import TensorFlowTaskRunner_v1  # NOQA
+    from .runner_tf import TensorFlowTaskRunner, TensorFlowTaskRunner_v1  # NOQA
     from .runner_keras import KerasTaskRunner  # NOQA
     from .fl_model import FederatedModel  # NOQA
 if importlib.util.find_spec('torch'):