diff --git a/gflownet/envs/crystals/ccrystal.py b/gflownet/envs/crystals/ccrystal.py
index c6eadb364..49cfdda8f 100644
--- a/gflownet/envs/crystals/ccrystal.py
+++ b/gflownet/envs/crystals/ccrystal.py
@@ -903,7 +903,7 @@ def states2proxy(
         states = tfloat(states, device=self.device, float_type=self.float)
         return torch.cat(
             [
-                subenv.states2oracle(self._get_states_of_subenv(states, stage))
+                subenv.states2proxy(self._get_states_of_subenv(states, stage))
                 for stage, subenv in self.subenvs.items()
             ],
             dim=1,