diff --git a/lib/main.dart b/lib/main.dart
index 9ad437f..d53b367 100644
--- a/lib/main.dart
+++ b/lib/main.dart
@@ -41,10 +41,12 @@ import 'package:flutter/foundation.dart';
 import 'package:responsive_framework/responsive_framework.dart';
 import 'package:shared_preferences/shared_preferences.dart';
 import 'package:sentry_flutter/sentry_flutter.dart';
+import 'package:flutter_driver/driver_extension.dart';
 
 const bool enableSentry = true;
 
 Future<void> main() async {
+  enableFlutterDriverExtension();
   WidgetsFlutterBinding.ensureInitialized();
 
   HomeProvider _homeProvider = HomeProvider();
diff --git a/test_driver/app_test.dart b/test_driver/app_test.dart
index 09de003..e9f4c76 100644
--- a/test_driver/app_test.dart
+++ b/test_driver/app_test.dart
@@ -113,6 +113,7 @@ void main() {
       await driver.enterText('triche');
       await tapOn('walletName');
       await driver.enterText(name);
+      await sleep(50);
       await tapOn('confirmStorage');
       await sleep(300);
       return pinCode;
@@ -439,8 +440,6 @@ void main() {
       await deleteAllWallets();
       await sleep(100);
       final String pincode = await createNewKeychain('Fast wallet');
-      await sleep(100);
-      await tapOn('manageWallets');
       await sleep(200);
       await driver.enterText(pincode);
       await sleep(100);