From 4e4d17b4cf1ef9ccae2976f30ffd4891a7716b13 Mon Sep 17 00:00:00 2001
From: James Williams <66931+jwill@users.noreply.github.com>
Date: Thu, 12 Sep 2024 15:16:43 -0400
Subject: [PATCH] Update color_screen_test.dart

---
 material_3_demo/test/color_screen_test.dart | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/material_3_demo/test/color_screen_test.dart b/material_3_demo/test/color_screen_test.dart
index 6282bf798..45d8e41eb 100644
--- a/material_3_demo/test/color_screen_test.dart
+++ b/material_3_demo/test/color_screen_test.dart
@@ -6,6 +6,7 @@ import 'package:flutter/material.dart';
 import 'package:flutter_test/flutter_test.dart';
 import 'package:material_3_demo/color_palettes_screen.dart';
 import 'package:material_3_demo/main.dart';
+import 'package:material_3_demo/scheme.dart';
 
 import 'component_screen_test.dart';
 
@@ -69,6 +70,6 @@ void main() {
     ));
     expect(find.text('Light ColorScheme'), findsOneWidget);
     expect(find.text('Dark ColorScheme'), findsOneWidget);
-    expect(find.byType(ColorGroup, skipOffstage: false), findsNWidgets(18));
+    expect(find.byType(SchemePreview, skipOffstage: false), findsNWidgets(2));
   });
 }