diff --git a/src/refiners/fluxion/layers/__init__.py b/src/refiners/fluxion/layers/__init__.py
index a48b747..4b55c64 100644
--- a/src/refiners/fluxion/layers/__init__.py
+++ b/src/refiners/fluxion/layers/__init__.py
@@ -24,6 +24,7 @@ from refiners.fluxion.layers.chain import (
     UseContext,
     SetContext,
     Parallel,
+    Distribute,
     Passthrough,
     Breakpoint,
     Concatenate,
@@ -71,6 +72,7 @@ __all__ = [
     "UseContext",
     "SetContext",
     "Parallel",
+    "Distribute",
     "Passthrough",
     "Breakpoint",
     "Concatenate",