diff --git a/examples/periodic_requests.py b/examples/periodic_requests.py
index cadbbe7..93fc35a 100755
--- a/examples/periodic_requests.py
+++ b/examples/periodic_requests.py
@@ -75,7 +75,7 @@ def on_device_status(status: DeviceStatus):
         print(f"\n--- Status Response #{status_count} ---")
         print(f"  Temperature: {status.dhwTemperature:.1f}°F")
         print(f"  Power: {status.currentInstPower:.1f}W")
-        print(f"  Available Energy: {status.availableEnergyCapacity:.1f}%")
+        print(f"  Available Energy: {status.availableEnergyCapacity:.0f} Wh")
 
     def on_device_feature(feature: DeviceFeature):
         """Callback receives parsed DeviceFeature objects."""
diff --git a/examples/tou_schedule_example.py b/examples/tou_schedule_example.py
index 3a4d2d6..a14a0bf 100644
--- a/examples/tou_schedule_example.py
+++ b/examples/tou_schedule_example.py
@@ -17,8 +17,13 @@ def capture_feature(feature) -> None:
         if not feature_future.done():
             feature_future.set_result(feature)
 
-    mqtt_client.once("feature_received", capture_feature)
+    # Subscribe to device feature messages first
+    await mqtt_client.subscribe_device_feature(device, capture_feature)
+
+    # Then request device info
     await mqtt_client.request_device_info(device)
+
+    # Wait for the response
     feature = await asyncio.wait_for(feature_future, timeout=15)
     return feature.controllerSerialNumber