diff --git a/ecoscope_workflows/tasks/__init__.py b/ecoscope_workflows/tasks/__init__.py
index 22ff38b0e..e6c98f273 100644
--- a/ecoscope_workflows/tasks/__init__.py
+++ b/ecoscope_workflows/tasks/__init__.py
@@ -2,3 +2,4 @@
 from . import io as io
 from . import preprocessing as preprocessing
 from . import results as results
+from . import transformation as transformation
diff --git a/ecoscope_workflows/tasks/transformation/__init__.py b/ecoscope_workflows/tasks/transformation/__init__.py
index e69de29bb..7d1085258 100644
--- a/ecoscope_workflows/tasks/transformation/__init__.py
+++ b/ecoscope_workflows/tasks/transformation/__init__.py
@@ -0,0 +1,5 @@
+from ._column_mapping import map_columns
+
+__all__ = [
+    "map_columns",
+]
diff --git a/ecoscope_workflows/tasks/transformation/column_mapping.py b/ecoscope_workflows/tasks/transformation/_column_mapping.py
similarity index 100%
rename from ecoscope_workflows/tasks/transformation/column_mapping.py
rename to ecoscope_workflows/tasks/transformation/_column_mapping.py