diff --git a/micropython/examples/cosmic_unicorn/cosmic_paint/cosmic_paint.py b/micropython/examples/cosmic_unicorn/cosmic_paint/cosmic_paint.py
index 2c006248..30a17f9e 100644
--- a/micropython/examples/cosmic_unicorn/cosmic_paint/cosmic_paint.py
+++ b/micropython/examples/cosmic_unicorn/cosmic_paint/cosmic_paint.py
@@ -1,6 +1,5 @@
 import os
-from microdot_asyncio import Microdot, send_file
-from microdot_asyncio_websocket import with_websocket
+from microdot import Microdot, send_file, websocket
 from phew import connect_to_wifi
 from cosmic import CosmicUnicorn
 from picographics import PicoGraphics, DISPLAY_COSMIC_UNICORN as DISPLAY
@@ -75,7 +74,7 @@ def flood_fill(x, y, r, g, b):
 
 
 @server.route('/paint')
-@with_websocket
+@websocket.with_websocket
 async def echo(request, ws):
     while True:
         data = await ws.receive()
diff --git a/micropython/examples/galactic_unicorn/galactic_paint/galactic_paint.py b/micropython/examples/galactic_unicorn/galactic_paint/galactic_paint.py
index a77ce6ce..91e034c8 100644
--- a/micropython/examples/galactic_unicorn/galactic_paint/galactic_paint.py
+++ b/micropython/examples/galactic_unicorn/galactic_paint/galactic_paint.py
@@ -1,6 +1,5 @@
 import os
-from microdot_asyncio import Microdot, send_file
-from microdot_asyncio_websocket import with_websocket
+from microdot import Microdot, send_file, websocket
 from phew import connect_to_wifi
 from galactic import GalacticUnicorn
 from picographics import PicoGraphics, DISPLAY_GALACTIC_UNICORN as DISPLAY
@@ -75,7 +74,7 @@ def flood_fill(x, y, r, g, b):
 
 
 @server.route('/paint')
-@with_websocket
+@websocket.with_websocket
 async def echo(request, ws):
     while True:
         data = await ws.receive()