1 files changed, 4 insertions, 3 deletions

diff --git a/examples/switch.rs b/examples/switch.rs
index 67767d7..0158ede 100644
--- a/examples/switch.rs
+++ b/examples/switch.rs
@@ -11,7 +11,7 @@ static RESOURCES: StaticCell<embassy_ha::DeviceResources> = StaticCell::new();
 async fn main_task(spawner: Spawner) {
     let mut stream = AsyncTcp::connect(std::env!("MQTT_ADDRESS"));
 
-    let mut device = embassy_ha::Device::new(
+    let mut device = embassy_ha::new(
         RESOURCES.init(Default::default()),
         embassy_ha::DeviceConfig {
             device_id: "example-device-id",
@@ -21,7 +21,8 @@ async fn main_task(spawner: Spawner) {
         },
     );
 
-    let switch = device.create_switch(
+    let switch = embassy_ha::create_switch(
+        &device,
         "switch-id",
         embassy_ha::SwitchConfig {
             common: embassy_ha::EntityCommonConfig {
@@ -34,7 +35,7 @@ async fn main_task(spawner: Spawner) {
 
     spawner.must_spawn(switch_task(switch));
 
-    device.run(&mut stream).await.unwrap();
+    embassy_ha::run(&mut device, &mut stream).await.unwrap();
 }
 
 #[embassy_executor::task]