1 files changed, 4 insertions, 3 deletions

diff --git a/examples/number.rs b/examples/number.rs
index 2231be1..8ef6656 100644
--- a/examples/number.rs
+++ b/examples/number.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 number = device.create_number(
+    let number = embassy_ha::create_number(
+        &device,
         "number-id",
         embassy_ha::NumberConfig {
             common: embassy_ha::EntityCommonConfig {
@@ -40,7 +41,7 @@ async fn main_task(spawner: Spawner) {
 
     spawner.must_spawn(number_task(number));
 
-    device.run(&mut stream).await.unwrap();
+    embassy_ha::run(&mut device, &mut stream).await.unwrap();
 }
 
 #[embassy_executor::task]