GDY game functions are triggering in a undesirable order about griddly HOT 2 OPEN

class ExtraStepHackSingleAgent(gym.Wrapper):
    def __init__(self, env, env_config):
        gym.Wrapper.__init__(self, env)
        self.action_space = env.action_space
        self.observation_space = env.observation_space
        # this doesn't seem to do anything...
        self.env.gdy.set_max_steps(2 * env_config.get('max_steps', 250))

    def step(self, action):
        ns, r, d, i = self.env.step(action)
        if not d:
            nns, rr, dd, ii = self.env.step(0)
            ii[-1] = {
                'ns': ns,
                'r': r,
                'd': d,
                'i': i
            }
            return nns, rr, dd, ii
        return ns, r, d, i

    def reset(self, **kwargs):
        return self.env.reset(**kwargs)