diff --git a/main.ts b/main.ts
index aa02ba0..8f5f3bd 100644
--- a/main.ts
+++ b/main.ts
@@ -1,9 +1,9 @@
 namespace ws2812b {
-    //% shim=sendBufferAsm
+    //% shim=light::sendWS2812Buffer
     export function sendBuffer(buf: Buffer, pin: DigitalPin) {
     }
 
-    //% shim=setBufferMode
+    //% shim=light::setMode
     export function setBufferMode(pin: DigitalPin, mode: number) {
 
     }
diff --git a/pxt.json b/pxt.json
index a71c4ea..c53c6e2 100644
--- a/pxt.json
+++ b/pxt.json
@@ -8,9 +8,7 @@
     },
     "files": [
         "README.md",
-        "main.ts",
-        "sendBuffer.asm",
-        "setBufferMode.asm"
+        "main.ts"
     ],
     "testFiles": [
         "test.ts"