diff --git a/tensorflow_graphics/geometry/convolution/BUILD b/tensorflow_graphics/geometry/convolution/BUILD
index 5de2e97d4..9b5f81056 100644
--- a/tensorflow_graphics/geometry/convolution/BUILD
+++ b/tensorflow_graphics/geometry/convolution/BUILD
@@ -27,6 +27,7 @@ py_library(
     # google internal rule 1
     deps = [
         ":graph_convolution",
+        ":graph_pooling",
         ":utils",
         "//tensorflow_graphics/util:export_api",
     ],
diff --git a/tensorflow_graphics/geometry/convolution/__init__.py b/tensorflow_graphics/geometry/convolution/__init__.py
index 69acd47a4..0879dd58a 100644
--- a/tensorflow_graphics/geometry/convolution/__init__.py
+++ b/tensorflow_graphics/geometry/convolution/__init__.py
@@ -17,6 +17,7 @@
 from __future__ import print_function
 
 from tensorflow_graphics.geometry.convolution import graph_convolution
+from tensorflow_graphics.geometry.convolution import graph_pooling
 from tensorflow_graphics.geometry.convolution import utils
 from tensorflow_graphics.util import export_api as _export_api