diff --git a/grain/python/__init__.py b/grain/python/__init__.py
index 0ac9d7d1f..1577dcfd4 100644
--- a/grain/python/__init__.py
+++ b/grain/python/__init__.py
@@ -37,6 +37,7 @@
     Batch,
     Filter as FilterTransform,
     Map as MapTransform,
+    FlatMap as FlatMapTransform,
     MapWithIndex as MapWithIndexTransform,
     RandomMap as RandomMapTransform,
     Transformation,