diff --git a/kmk/extensions/oled.py b/kmk/extensions/oled.py
index 34e61f7..cf8eccf 100644
--- a/kmk/extensions/oled.py
+++ b/kmk/extensions/oled.py
@@ -114,7 +114,7 @@ class Oled(Extension):
     def on_runtime_disable(self, sandbox):
         return
 
-    def during_bootup(self, board):
+    def during_bootup(self, keyboard):
         displayio.release_displays()
         i2c = busio.I2C(keyboard.SCL, keyboard.SDA)
         self._display = adafruit_displayio_ssd1306.SSD1306(