From 002153cab23d423fce0822e27b4a3845f89eeda4 Mon Sep 17 00:00:00 2001
From: Cecilia Ferrari <cecilia.ferrari@gssi.it>
Date: Mon, 22 Jan 2024 16:03:38 +0100
Subject: [PATCH] import new block

---
 flamedisx/__init__.py    | 1 +
 flamedisx/lxe_sources.py | 2 ++
 2 files changed, 3 insertions(+)

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)