From 29838abb403206c5e0707194f1ec665331fe4ca0 Mon Sep 17 00:00:00 2001
From: Lucas Scharenbroich <lscharen@prowestgis.com>
Date: Tue, 27 Jun 2023 21:30:30 -0500
Subject: [PATCH] Shorter size-agnostic implemention of SetDataBank

---
 src/Tool.s | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/src/Tool.s b/src/Tool.s
index ded21ad..d5c5121 100644
--- a/src/Tool.s
+++ b/src/Tool.s
@@ -119,11 +119,16 @@ _GTERemoveSprite     MAC
                      UserTool  $1300+GTEToolNum
                      <<<
 ; Helper function to set the data back to the toolset default
-_SetDataBank    sep  #$20
-                lda  #^TileStore
-                pha
+;_SetDataBank    sep  #$20
+;                lda  #^TileStore
+;                pha
+;                plb
+;                rep  #$20
+;                rts
+
+_SetDataBank    pea  #>TileStore
+                plb
                 plb
-                rep  #$20
                 rts
 
 ; Do nothing when the tool set is installed