diff --git a/ui/storybook/stories/data-viz-misc.stories.tsx b/ui/storybook/stories/data-viz-misc.stories.tsx
index d3474ee9a30..2022b3cd43a 100644
--- a/ui/storybook/stories/data-viz-misc.stories.tsx
+++ b/ui/storybook/stories/data-viz-misc.stories.tsx
@@ -317,10 +317,15 @@ function LiveRunWidgetStory({ empty = false, loading = false }: { empty?: boolea
 
 function OpenOnboardingOnMount({ initialStep }: { initialStep: 1 | 2 }) {
   const { openOnboarding } = useDialog();
+  const queryClient = useQueryClient();
 
   useEffect(() => {
+    queryClient.setQueryData(queryKeys.agents.adapterModels(companyId, "claude_local"), [
+      { id: "claude-sonnet-4-5", label: "Claude Sonnet 4.5" },
+      { id: "claude-opus-4-1", label: "Claude Opus 4.1" },
+    ]);
     openOnboarding(initialStep === 1 ? { initialStep } : { initialStep, companyId });
-  }, [initialStep, openOnboarding]);
+  }, [initialStep, openOnboarding, queryClient]);
 
   return <OnboardingWizard />;
 }
diff --git a/ui/storybook/stories/dialogs-modals.stories.tsx b/ui/storybook/stories/dialogs-modals.stories.tsx
index e43910db02b..43947841174 100644
--- a/ui/storybook/stories/dialogs-modals.stories.tsx
+++ b/ui/storybook/stories/dialogs-modals.stories.tsx
@@ -1,4 +1,4 @@
-import { useEffect, useLayoutEffect, useState, type ReactNode } from "react";
+import { useEffect, useLayoutEffect, useRef, useState, type ReactNode } from "react";
 import type { Meta, StoryObj } from "@storybook/react-vite";
 import type {
   DocumentRevision,
@@ -468,7 +468,10 @@ function clickButtonByText(text: string) {
 }
 
 function useOpenOnMount(open: () => void) {
-  useEffect(() => {
+  const didOpenRef = useRef(false);
+  useLayoutEffect(() => {
+    if (didOpenRef.current) return;
+    didOpenRef.current = true;
     open();
   }, [open]);
 }