diff --git a/pcdet/models/backbones_3d/pfe/__init__.py b/pcdet/models/backbones_3d/pfe/__init__.py
new file mode 100644
index 0000000..b65a3f5
--- /dev/null
+++ b/pcdet/models/backbones_3d/pfe/__init__.py
@@ -0,0 +1,5 @@
+from .voxel_set_abstraction import VoxelSetAbstraction
+
+__all__ = {
+    'VoxelSetAbstraction': VoxelSetAbstraction
+}