diff --git a/adafruit_displayio_ssd1306.py b/adafruit_displayio_ssd1306.py
index 13edd14..3e67cfe 100644
--- a/adafruit_displayio_ssd1306.py
+++ b/adafruit_displayio_ssd1306.py
@@ -31,13 +31,15 @@
   https://github.com/adafruit/circuitpython/releases
 
 """
-
-import displayio
-
 try:
     from typing import Union
+    from busdisplay import BusDisplay
+    from fourwire import FourWire
+    from i2cdisplaybus import I2CDisplayBus
 except ImportError:
-    pass
+    from displayio import FourWire
+    from displayio import I2CDisplay as I2CDisplayBus
+    from displayio import Display as BusDisplay
 
 __version__ = "0.0.0+auto.0"
 __repo__ = "https://github.com/adafruit/Adafruit_CircuitPython_DisplayIO_SSD1306.git"
@@ -60,7 +62,7 @@
 )
 
 
-class SSD1306(displayio.Display):
+class SSD1306(BusDisplay):
     """
     SSD1306 driver
 
@@ -70,9 +72,7 @@ class SSD1306(displayio.Display):
         (0, 90, 180, 270)
     """
 
-    def __init__(
-        self, bus: Union[displayio.FourWire, displayio.I2CDisplay], **kwargs
-    ) -> None:
+    def __init__(self, bus: Union[FourWire, I2CDisplayBus], **kwargs) -> None:
         # Patch the init sequence for 32 pixel high displays.
         init_sequence = bytearray(_INIT_SEQUENCE)
         height = kwargs["height"]