diff --git a/pearl/utils/instantiations/environments/__init__.py b/pearl/utils/instantiations/environments/__init__.py
index ad216a29..6ada4058 100644
--- a/pearl/utils/instantiations/environments/__init__.py
+++ b/pearl/utils/instantiations/environments/__init__.py
@@ -10,9 +10,8 @@
 )
 from .contextual_bandit_uci_environment import SLCBEnvironment
 from .environments import (
-    BoxObservationsEnvironmentBase,
-    BoxObservationsFromDiscrete,
     FixedNumberOfStepsEnvironment,
+    ObservationTransformationEnvironmentAdapterBase,
     OneHotObservationsFromDiscrete,
 )
 from .gym_environment import GymEnvironment
@@ -27,8 +26,7 @@
 
 
 __all__ = [
-    "BoxObservationsEnvironmentBase",
-    "BoxObservationsFromDiscrete",
+    "ObservationTransformationEnvironmentAdapterBase",
     "ContinuousSparseRewardEnvironment",
     "ContextualBanditEnvironment",
     "ContextualBanditLinearSyntheticEnvironment",