diff --git a/packages/design-system/src/stories/components/Dialog/Dialog.stories.tsx b/packages/design-system/src/stories/components/Dialog/Dialog.stories.tsx
index bc6c920e..8800fc02 100644
--- a/packages/design-system/src/stories/components/Dialog/Dialog.stories.tsx
+++ b/packages/design-system/src/stories/components/Dialog/Dialog.stories.tsx
@@ -94,7 +94,7 @@ export const ActionModal: Story = {
             </>
           }
         >
-          <DialogTitle title="Title area" onClose={close} />
+          <DialogTitle title="Title area" />
           <DialogContent>
             Lorem Ipsum is simply dummy text of the a printing and typesetting
             industry
@@ -114,6 +114,10 @@ export const NonModal: Story = {
       setIsOpen(true);
     }
 
+    function close() {
+      setIsOpen(false);
+    }
+
     return (
       <>
         <Box sx={{ display: "flex", gap: "8px" }}>
@@ -146,7 +150,7 @@ export const NonModal: Story = {
             </>
           }
         >
-          <DialogTitle title="Title area" onClose={close} />
+          <DialogTitle title="Title area" />
           <DialogContent>
             Lorem Ipsum is simply dummy text of the a printing and typesetting
             industry