diff --git a/src/gfn/modules.py b/src/gfn/modules.py
index 1dde4fa4..13080c7f 100644
--- a/src/gfn/modules.py
+++ b/src/gfn/modules.py
@@ -81,7 +81,7 @@ def forward(self, states: States) -> TT["batch_shape", "output_dim", float]:
         return out
 
     def __repr__(self):
-        return f"{self.__class__.__name__}({self.env})"
+        return f"{self.__class__.__name__} module"
 
     @property
     @abstractmethod