fix(battery_level_trait_provider.dart): Now correctly report the batt…

…ery level
Yonomi · Dec 21, 2021 · 6cf915e · 6cf915e
1 parent 5a0d398
commit 6cf915e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/lib/providers/battery_level_trait_provider.dart b/lib/providers/battery_level_trait_provider.dart
@@ -57,7 +57,9 @@ class BatteryLevelTraitProvider extends ChangeNotifier {
 
   BatteryLevelTrait? getBatteryLevelTrait() {
     try {
-      return _deviceDetail?.traits.first as BatteryLevelTrait?;
+      return _deviceDetail?.traits
+              .firstWhere((trait) => trait.runtimeType == BatteryLevelTrait)
+          as BatteryLevelTrait?;
     } catch (error) {
       return null;
     }