diff --git a/shimmer/__init__.py b/shimmer/__init__.py
index b2699cd2..124065a2 100644
--- a/shimmer/__init__.py
+++ b/shimmer/__init__.py
@@ -19,6 +19,7 @@
 from shimmer.modules.gw_module import (
     GWDecoder,
     GWEncoder,
+    GWEncoderLinear,
     GWModule,
     GWModuleBase,
     VariationalGWEncoder,
@@ -65,6 +66,7 @@
     "DomainModule",
     "GWDecoder",
     "GWEncoder",
+    "GWEncoderLinear",
     "VariationalGWEncoder",
     "GWModuleBase",
     "GWModule",
diff --git a/shimmer/modules/__init__.py b/shimmer/modules/__init__.py
index 698ed87a..bcd0a6af 100644
--- a/shimmer/modules/__init__.py
+++ b/shimmer/modules/__init__.py
@@ -19,6 +19,7 @@
 from shimmer.modules.gw_module import (
     GWDecoder,
     GWEncoder,
+    GWEncoderLinear,
     GWModule,
     GWModuleBase,
     VariationalGWEncoder,
@@ -51,6 +52,7 @@
     "DomainModule",
     "GWDecoder",
     "GWEncoder",
+    "GWEncoderLinear",
     "VariationalGWEncoder",
     "GWModuleBase",
     "GWModule",