diff --git a/examples/arduino-zigbee-switch/src/Zigbee_On_Off_Switch.ino b/examples/arduino-zigbee-switch/src/Zigbee_On_Off_Switch.ino
index 09487de..69cf665 100644
--- a/examples/arduino-zigbee-switch/src/Zigbee_On_Off_Switch.ino
+++ b/examples/arduino-zigbee-switch/src/Zigbee_On_Off_Switch.ino
@@ -31,8 +31,7 @@
 #error "Zigbee coordinator mode is not selected in Tools->Zigbee mode"
 #endif
 
-#include "ZigbeeCore.h"
-#include "ep/ZigbeeSwitch.h"
+#include "Zigbee.h"
 
 #define SWITCH_ENDPOINT_NUMBER 5