diff --git a/src/xminigrid/wrappers.py b/src/xminigrid/wrappers.py
index 417c864..6f1b97b 100644
--- a/src/xminigrid/wrappers.py
+++ b/src/xminigrid/wrappers.py
@@ -74,6 +74,8 @@ def step(self, params, timestep, action):
 # Yes, these are a bit stupid, but a tmp workaround to not write an actual system for spaces.
 # May be, in the future, I will port the entire API to some existing one, like functional Gymnasium.
 # For now, faster to do this stuff with dicts instead...
+# NB: if you do not want to use this (due to the dicts as obs),
+# just get needed parts from the original TimeStep and State dataclasses
 class DirectionObsWrapper(Wrapper):
     def observation_shape(self, params):
         base_shape = self._env.observation_shape(params)