diff --git a/packages/integration_test/README.md b/packages/integration_test/README.md
index 7ba3a031c3..305fad3241 100644
--- a/packages/integration_test/README.md
+++ b/packages/integration_test/README.md
@@ -140,7 +140,7 @@ import 'package:integration_test/integration_test_driver_extended.dart';
 
 Future<void> main() async {
   await integrationDriver(
-    onScreenshot: (String screenshotName, List<int> screenshotBytes) async {
+    onScreenshot: (String screenshotName, List<int> screenshotBytes, [Map<String, Object?>? args]) async {
       final File image = File('$screenshotName.png');
       image.writeAsBytesSync(screenshotBytes);
       // Return false if the screenshot is invalid.