diff --git a/test/bug_fix_test.dart b/test/bug_fix_test.dart
index 5867f67c..fb135f12 100644
--- a/test/bug_fix_test.dart
+++ b/test/bug_fix_test.dart
@@ -16,15 +16,17 @@ void main() {
 
         await tester.pumpWidget(
           MaterialApp(
-            home: QuillSimpleToolbar(
-              configurations: QuillSimpleToolbarConfigurations(
-                controller: controller,
-                showRedo: false,
-                customButtons: const [
-                  QuillToolbarCustomButtonOptions(
-                    tooltip: tooltip,
-                  )
-                ],
+            home: Scaffold(
+              body: QuillSimpleToolbar(
+                configurations: QuillSimpleToolbarConfigurations(
+                  controller: controller,
+                  showRedo: false,
+                  customButtons: const [
+                    QuillToolbarCustomButtonOptions(
+                      tooltip: tooltip,
+                    )
+                  ],
+                ),
               ),
             ),
           ),