diff --git a/pokemonred_puffer/environment.py b/pokemonred_puffer/environment.py index 412813b..7ca9738 100644 --- a/pokemonred_puffer/environment.py +++ b/pokemonred_puffer/environment.py @@ -583,7 +583,7 @@ def step(self, action): if self.save_video and self.step_count == 0: self.start_video() - _, wMapPalOffset = self.pyboy.symbol_lookup("wMapPalOffset")[1] + _, wMapPalOffset = self.pyboy.symbol_lookup("wMapPalOffset") if self.auto_flash and self.pyboy.memory[wMapPalOffset] == 6: self.pyboy.memory[wMapPalOffset] = 0