diff --git a/lib/src/ndialogBase.dart b/lib/src/ndialogBase.dart
index b43e11d..100f9cd 100644
--- a/lib/src/ndialogBase.dart
+++ b/lib/src/ndialogBase.dart
@@ -30,7 +30,7 @@ class NDialog extends StatelessWidget {
   @override
   Widget build(BuildContext context) {
     final ThemeData? theme = Theme.of(context);
-    final DialogTheme? dialogTheme = DialogTheme.of(context);
+    final DialogThemeData dialogTheme = DialogTheme.of(context);
     final DialogStyle style = dialogStyle ?? DialogStyle();
 
     String? label = style.semanticsLabel;
diff --git a/lib/src/progressDialog.dart b/lib/src/progressDialog.dart
index 37de1ec..3baa7ba 100644
--- a/lib/src/progressDialog.dart
+++ b/lib/src/progressDialog.dart
@@ -289,7 +289,7 @@ class _ProgressDialogWidgetState extends State<_ProgressDialogWidget>
   @override
   Widget build(BuildContext context) {
     final ThemeData theme = Theme.of(context);
-    final DialogTheme dialogTheme = DialogTheme.of(context);
+    final DialogThemeData dialogTheme = DialogTheme.of(context);
 
     Widget? title = _title ?? widget.title;
     Widget message = _message ?? (widget.message ?? SizedBox.shrink());