diff --git a/flytekit/extras/tensorflow/__init__.py b/flytekit/extras/tensorflow/__init__.py
index 1bbd9bbe11..0c4d12f443 100644
--- a/flytekit/extras/tensorflow/__init__.py
+++ b/flytekit/extras/tensorflow/__init__.py
@@ -26,8 +26,8 @@
 
 
 if _tensorflow_installed:
-    from .record import TensorFlowRecordFileTransformer, TensorFlowRecordsDirTransformer
     from .model import TensorFlowModelTransformer
+    from .record import TensorFlowRecordFileTransformer, TensorFlowRecordsDirTransformer
 else:
     logger.info(
         "We won't register TensorFlowRecordFileTransformer, TensorFlowRecordsDirTransformer and TensorFlowModelTransformer"