diff --git a/src/Kaleidoscope-Model01-TestMode.h b/src/Kaleidoscope-Model01-TestMode.h
index b2ec7aed..883fa2aa 100644
--- a/src/Kaleidoscope-Model01-TestMode.h
+++ b/src/Kaleidoscope-Model01-TestMode.h
@@ -12,7 +12,9 @@ class TestMode_ : public KaleidoscopePlugin {
     static void run_tests();
     static void test_leds();
     static void test_matrix();
+    static void wait_for_keypress();
     static void loopHook(bool postClear);
+    static void set_leds(uint8_t r, uint8_t g, uint8_t b);
 };
 
 extern TestMode_ TestMode;