def reset(self, state=None):
        env_mask = np.zeros(self._n_envs, dtype=bool)
        env_mask[self._default_env] = True
        return self.reset_all(env_mask, state)