diff --git a/flamedisx/__init__.py b/flamedisx/__init__.py
index 3aac816d3..ad266a2d5 100644
--- a/flamedisx/__init__.py
+++ b/flamedisx/__init__.py
@@ -16,6 +16,7 @@
 from .lxe_blocks.energy_spectrum import *
 from .lxe_blocks.quanta_generation import *
 from .lxe_blocks.quanta_splitting import *
+from .lxe_blocks.electron_wall_loss import *
 from .lxe_blocks.detection import *
 from .lxe_blocks.double_pe import *
 from .lxe_blocks.raw_signals import *
diff --git a/flamedisx/lxe_sources.py b/flamedisx/lxe_sources.py
index b8bc1ee25..1ec7dbb1a 100644
--- a/flamedisx/lxe_sources.py
+++ b/flamedisx/lxe_sources.py
@@ -14,6 +14,7 @@ class ERSource(fd.BlockModelSource):
         fd.MakeS1Photoelectrons,
         fd.MakeS1,
         fd.ReconstructS1,
+        fd.ElectronWallLoss,
         fd.DetectElectrons,
         fd.MakeS2,
         fd.ReconstructS2)
@@ -48,6 +49,7 @@ class NRSource(fd.BlockModelSource):
         fd.MakeS1Photoelectrons,
         fd.MakeS1,
         fd.ReconstructS1,
+        fd.ElectronWallLoss,
         fd.DetectElectrons,
         fd.MakeS2,
         fd.ReconstructS2)