diff --git a/sw/sevenseg.s b/sw/sevenseg.s
index e915d59..6b9efc6 100644
--- a/sw/sevenseg.s
+++ b/sw/sevenseg.s
@@ -18,6 +18,7 @@ _hex_set_8:
         phx
         cmp #$3         ; If idx >= 3 then fail
         bcc @1
+        plx
         lda #$1
         rts
 @1:     tax             ; Move idx into x
@@ -44,6 +45,7 @@ _hex_set_24:
         stx SEVEN_SEG+1
         lda sreg
         sta SEVEN_SEG+2
+        lda #$0
         rts
 
 ; @in A: mask