Refresh the screen when receiving a change from the host

1 files changed, 9 insertions, 1 deletions

diff --git a/oled.py b/oled.py
index 4db0436..ab4c917 100644
--- a/oled.py
+++ b/oled.py
@@ -34,6 +34,7 @@ class CustomDisplay(Display):
         layout is updated
     """
     def __init__(self, *args, **kwargs):
+        self.keyboard = None
         super().__init__(*args, **kwargs)
         self._prevLayers = None
 
@@ -47,12 +48,19 @@ class CustomDisplay(Display):
         """
         super().before_matrix_scan(sandbox)
         if self.keyboard.keymap[0].name != self._prevLayers:
-            self.timer_start = ticks_ms()
             self._prevLayers = self.keyboard.keymap[0].name
             self.render(sandbox.active_layers[0])
         return
 
+    @property
+    def timer_start(self):
+        if self.keyboard:
+            return self.keyboard.keymap[0].timer_start
 
+    @timer_start.setter
+    def timer_start(self, value):
+        if self.keyboard:
+            self.keyboard.keymap[0].timer_start = value
 
 def main(keyboard):