diff --git a/scribe/lib/scribe/ai/presentation/widgets/button/inline_ai_assist_button.dart b/scribe/lib/scribe/ai/presentation/widgets/button/inline_ai_assist_button.dart
index d42d68605..d47193cf4 100644
--- a/scribe/lib/scribe/ai/presentation/widgets/button/inline_ai_assist_button.dart
+++ b/scribe/lib/scribe/ai/presentation/widgets/button/inline_ai_assist_button.dart
@@ -43,12 +43,10 @@ class InlineAiAssistButton extends StatelessWidget {
       size = renderBox.size;
     }
 
-    await onTapFallback?.call();
-
-    if (!context.mounted) return;
-
     final isScribeMobile = AiScribeMobileUtils.isScribeInMobileMode(context);
 
+    await onTapFallback?.call();
+
     await AiScribeModalManager.showAIScribeModal(
       isScribeMobile: isScribeMobile,
       imagePaths: imagePaths,