diff --git a/lib/src/services/run_session.dart b/lib/src/services/run_session.dart
index 6ed8ab95..0a9d7bd9 100644
--- a/lib/src/services/run_session.dart
+++ b/lib/src/services/run_session.dart
@@ -8,7 +8,7 @@ import 'package:mdigit_span_tasks_ema/src/services/data_processor.dart';
 /// Run a data collection session
 /// Running a session includes configuring everything needed and running a
 /// cognitive task specified with [taskRunner].
-void runSession(
+Future<void> runSession(
     {required Function({
       required String participantID,
       required String sessionID,
diff --git a/lib/src/task_list/view/task_buttons.dart b/lib/src/task_list/view/task_buttons.dart
index f52d2555..32c0531f 100644
--- a/lib/src/task_list/view/task_buttons.dart
+++ b/lib/src/task_list/view/task_buttons.dart
@@ -11,8 +11,10 @@ class DSBButton extends StatelessWidget {
   @override
   Widget build(BuildContext context) {
     return ElevatedButton(
-      onPressed: () => runSession(
-          taskRunner: runDigitSpanBackwards, taskName: 'ds_backwards'),
+      onPressed: () async {
+        await runSession(
+            taskRunner: runDigitSpanBackwards, taskName: 'ds_backwards');
+      },
       child: Text(
         'Digit Span Backwards',
         style: Theme.of(context).textTheme.titleLarge,
@@ -29,8 +31,10 @@ class DSFButton extends StatelessWidget {
   @override
   Widget build(BuildContext context) {
     return ElevatedButton(
-      onPressed: () =>
-          runSession(taskRunner: runDigitSpanForward, taskName: 'ds_forward'),
+      onPressed: () async {
+        await runSession(
+            taskRunner: runDigitSpanForward, taskName: 'ds_forward');
+      },
       child: Text(
         'Digit Span Forward',
         style: Theme.of(context).textTheme.titleLarge,