From b5cf50b8191a2770889ca619ef2fdce0ded8f719 Mon Sep 17 00:00:00 2001
From: "regicidal.plutophage"
 <36969337+regicidalplutophage@users.noreply.github.com>
Date: Wed, 1 Mar 2023 09:42:46 +0300
Subject: [PATCH] Update oled.py

---
 kmk/extensions/oled.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/kmk/extensions/oled.py b/kmk/extensions/oled.py
index 9d83a93..fcb7de3 100644
--- a/kmk/extensions/oled.py
+++ b/kmk/extensions/oled.py
@@ -147,8 +147,8 @@ class Oled(Extension):
             self.updateOLED(sandbox)
         return
 
-    def after_matrix_scan(self, keyboard):
-        if keyboard.matrix_update or keyboard.secondary_matrix_update:
+    def after_matrix_scan(self, sandbox):
+        if sandbox.matrix_update or sandbox.secondary_matrix_update:
             self.timer_time_reset()
             return
 
@@ -197,4 +197,4 @@ class Oled(Extension):
             if self._display.brightness > 0.5:
                 self._display.brightness = 0.5
         else: self._display.brightness = (self._brightness)
-        return
\ No newline at end of file
+        return