diff --git a/src/cosmosmesh_privacy_preserving_federated/energi_bridge.py b/src/cosmosmesh_privacy_preserving_federated/energi_bridge.py
index beeaffe..088f341 100644
--- a/src/cosmosmesh_privacy_preserving_federated/energi_bridge.py
+++ b/src/cosmosmesh_privacy_preserving_federated/energi_bridge.py
@@ -101,4 +101,4 @@ class EnergiBridge:
 
 
 
-__all__ = ["EnergiBridge", "EnergiLocalProblem", "EnergiSignal"]
+__all__ = ["EnergiBridge", "EnergiSignal"]