diff --git a/dev-requirements.txt b/dev-requirements.txt
index 102fa714553..815158586bf 100644
--- a/dev-requirements.txt
+++ b/dev-requirements.txt
@@ -2,7 +2,7 @@
 # This file is autogenerated by pip-compile with python 3.9
 # To update, run:
 #
-#    pip-compile --output-file=dev-requirements.txt dev-requirements.in
+#    make dev-requirements.txt
 #
 -e file:.#egg=flytekit
     # via
diff --git a/flytekit/extras/tensorflow/__init__.py b/flytekit/extras/tensorflow/record/__init__.py
similarity index 100%
rename from flytekit/extras/tensorflow/__init__.py
rename to flytekit/extras/tensorflow/record/__init__.py
diff --git a/flytekit/extras/tensorflow/records.py b/flytekit/extras/tensorflow/record/records.py
similarity index 100%
rename from flytekit/extras/tensorflow/records.py
rename to flytekit/extras/tensorflow/record/records.py
diff --git a/tests/flytekit/unit/extras/tensorflow/test_transformations.py b/tests/flytekit/unit/extras/tensorflow/test_transformations.py
index d5d2a8a2719..4beb585b408 100644
--- a/tests/flytekit/unit/extras/tensorflow/test_transformations.py
+++ b/tests/flytekit/unit/extras/tensorflow/test_transformations.py
@@ -4,7 +4,7 @@
 import flytekit
 from flytekit.configuration import Image, ImageConfig
 from flytekit.core import context_manager
-from flytekit.extras.tensorflow import TensorflowExampleTransformer
+from flytekit.extras.tensorflow.record import TensorflowExampleTransformer
 from flytekit.models.core.types import BlobType
 from flytekit.models.literals import BlobMetadata
 from flytekit.models.types import LiteralType