diff --git a/test_driver/screenshot_driver.dart b/test_driver/screenshot_driver.dart
index 476709ee..f98d10c7 100644
--- a/test_driver/screenshot_driver.dart
+++ b/test_driver/screenshot_driver.dart
@@ -6,7 +6,7 @@ import 'package:integration_test/integration_test_driver_extended.dart';
 Future<void> main() async {
   try {
     await integrationDriver(
-      onScreenshot: (String screenshotName, List<int> screenshotBytes) async {
+      onScreenshot: (String screenshotName, List<int> screenshotBytes, [_]) async {
         final File image = await File(screenshotName).create(recursive: true);
         image.writeAsBytesSync(screenshotBytes);
         return true;