diff --git a/packages/ui/stories/components/primitives/Sonner.stories.tsx b/packages/ui/stories/components/primitives/Sonner.stories.tsx
index 207a12f64b..619e562952 100644
--- a/packages/ui/stories/components/primitives/Sonner.stories.tsx
+++ b/packages/ui/stories/components/primitives/Sonner.stories.tsx
@@ -206,17 +206,15 @@ export const Warning: Story = {
 
 export const Loading: Story = {
   render: () => {
-    const mockPromise = new Promise((resolve) => {
-      setTimeout(() => resolve('Data loaded'), 2000)
-    })
-
     return (
       <div className="flex flex-col gap-3">
         <Button
           onClick={() =>
             toast.loading('Loading...', {
               description: 'Please wait while we process your request.',
-              promise: mockPromise
+              promise: new Promise((resolve) => {
+                setTimeout(() => resolve('Data loaded'), 2000)
+              })
             })
           }>
           Show Loading Toast