small tweak in gridworld