diff --git a/src/main/python/firmware_flasher.py b/src/main/python/firmware_flasher.py
index 5be3c8e..a1bc133 100644
--- a/src/main/python/firmware_flasher.py
+++ b/src/main/python/firmware_flasher.py
@@ -156,8 +156,10 @@ class FirmwareFlasher(BasicEditor):
 
         if isinstance(self.device, VialBootloader):
             self.log("Valid Vial Bootloader device at {}".format(self.device.desc["path"].decode("utf-8")))
+            self.chk_restore_keymap.hide()
         elif isinstance(self.device, VialKeyboard):
             self.log("Vial keyboard detected")
+            self.chk_restore_keymap.show()
 
     def valid(self):
         return isinstance(self.device, VialBootloader) or\