diff --git a/examples/neokey1x4_simpletest.py b/examples/neokey1x4_simpletest.py
index bad1927..2ef8faa 100644
--- a/examples/neokey1x4_simpletest.py
+++ b/examples/neokey1x4_simpletest.py
@@ -1,6 +1,6 @@
 # SPDX-FileCopyrightText: 2021 ladyada for Adafruit Industries
 # SPDX-License-Identifier: MIT
-
+"""NeoKey simpletest."""
 import board
 from adafruit_neokey.neokey1x4 import NeoKey1x4
 
diff --git a/examples/neokey_1x4_multikey.py b/examples/neokey_1x4_multikey.py
index 5bc8fa4..0b2f5a8 100644
--- a/examples/neokey_1x4_multikey.py
+++ b/examples/neokey_1x4_multikey.py
@@ -2,13 +2,9 @@
 # SPDX-License-Identifier: MIT
 """Example for connecting two NeoKey 1x4 breakouts. Requires bridging the A0 jumper on one board."""
 import board
+from rainbowio import colorwheel
 from adafruit_neokey.neokey1x4 import NeoKey1x4
 
-try:
-    from _pixelbuf import colorwheel
-except ImportError:
-    from adafruit_pypixelbuf import colorwheel
-
 # Create a NeoKey object
 neokey1 = NeoKey1x4(board.I2C())
 neokey2 = NeoKey1x4(board.I2C(), addr=0x31)