diff --git a/dev/integration_tests/flutter_gallery/lib/demo/material/dialog_demo.dart b/dev/integration_tests/flutter_gallery/lib/demo/material/dialog_demo.dart
index 04a2c13b5fd8..764670166ca0 100644
--- a/dev/integration_tests/flutter_gallery/lib/demo/material/dialog_demo.dart
+++ b/dev/integration_tests/flutter_gallery/lib/demo/material/dialog_demo.dart
@@ -71,7 +71,7 @@ class DialogDemoState extends State<DialogDemo> {
       builder: (BuildContext context) => child!,
     )
     .then((T? value) { // The value passed to Navigator.pop() or null.
-      if (value != null) {
+      if (context.mounted && value != null) {
         ScaffoldMessenger.of(context).showSnackBar(SnackBar(
           content: Text('You selected: $value'),
         ));
@@ -179,6 +179,9 @@ class DialogDemoState extends State<DialogDemo> {
                 initialTime: _selectedTime!,
               )
               .then((TimeOfDay? value) {
+                if (!mounted) {
+                  return;
+                }
                 if (value != null && value != _selectedTime) {
                   _selectedTime = value;
                   ScaffoldMessenger.of(context).showSnackBar(SnackBar(