diff --git a/example/integration_test/app_test.dart b/example/integration_test/app_test.dart
index 47a6340..c56c4e8 100644
--- a/example/integration_test/app_test.dart
+++ b/example/integration_test/app_test.dart
@@ -1,5 +1,4 @@
-import 'package:flutter/foundation.dart'
-    show kIsWeb, defaultTargetPlatform, TargetPlatform;
+import 'package:flutter/foundation.dart';
 import 'package:flutter/services.dart';
 import 'package:flutter_test/flutter_test.dart';
 import 'package:image/image.dart' as img;
diff --git a/example/test_driver/integration_test.dart b/example/test_driver/integration_test.dart
index b38629c..172b4b7 100644
--- a/example/test_driver/integration_test.dart
+++ b/example/test_driver/integration_test.dart
@@ -1,3 +1,5 @@
 import 'package:integration_test/integration_test_driver.dart';
 
-Future<void> main() => integrationDriver();
+void main() async {
+  await integrationDriver();
+}