1 files changed, 8 insertions, 4 deletions
diff --git a/lib/widgets/level_list_entry.dart b/lib/widgets/level_list_entry.dart
index a47a9c2..832186f 100644
--- a/lib/widgets/level_list_entry.dart
+++ b/lib/widgets/level_list_entry.dart
@@ -25,10 +25,14 @@ class LevelListEntry extends StatelessWidget {
       context: context,
       builder: (BuildContext context) {
         return ESenseConnectDialog(
-            deviceStatus: ESenseInput.instance.deviceStatus,
-            connect: (String name) {
-              ESenseInput.instance.connectToESense(name);
-            });
+          deviceStatus: ESenseInput.instance.deviceStatus,
+          connect: (String name) {
+            ESenseInput.instance.connectToESense(name);
+          },
+          disconnect: () {
+            ESenseInput.instance.eSenseManager.disconnect();
+          },
+        );
       },
     );
   }