summaryrefslogtreecommitdiff
path: root/pokemontools/vba/vba.py
diff options
context:
space:
mode:
Diffstat (limited to 'pokemontools/vba/vba.py')
-rw-r--r--pokemontools/vba/vba.py2
1 files changed, 2 insertions, 0 deletions
diff --git a/pokemontools/vba/vba.py b/pokemontools/vba/vba.py
index 10513c6..02fd99d 100644
--- a/pokemontools/vba/vba.py
+++ b/pokemontools/vba/vba.py
@@ -1073,6 +1073,8 @@ class crystal(object):
if not wram:
self.inject_script_into_rom(asm=script, wram_address=address)
else:
+ # TODO: move this into a separate function. Maybe use a context
+ # manager to restore wram at the end.
mem = list(self.vba.memory)
backup_wram = mem[address : address + len(script)]
mem[address : address + len(script)] = script