From 06cbcd4fa9b13c81dfb9c748843b40c05381c074 Mon Sep 17 00:00:00 2001
From: Joshua Bell <inexorabletash@gmail.com>
Date: Sat, 28 Sep 2019 09:41:16 -0700
Subject: [PATCH] Restore SSC flags after detection. Fixes #8

---
 cricket.system.s | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/cricket.system.s b/cricket.system.s
index 00fc603..e89164f 100644
--- a/cricket.system.s
+++ b/cricket.system.s
@@ -215,13 +215,11 @@ digit:  cmp     #HI('0')          ; < '0' ?
         bcc     :-
 
 cricket_found:
+        jsr     restore_cmd_ctl
         jmp     install_driver
 
 cricket_not_found:
-        lda     saved_control
-        sta     CONTROL
-        lda     saved_command
-        sta     COMMAND
+        jsr     restore_cmd_ctl
         ;; fall through...
 
 not_found:
@@ -231,6 +229,13 @@ not_found:
         HIASCIIZ CR, CR, CR, PRODUCT, " - Not Found."
         jmp     launch_next_sys_file
 
+restore_cmd_ctl:
+        lda     saved_control
+        sta     CONTROL
+        lda     saved_command
+        sta     COMMAND
+        rts
+
 saved_command:  .byte   0
 saved_control:  .byte   0
 .endproc