diff --git a/packages/react/stories/Dialog.stories.tsx b/packages/react/stories/Dialog.stories.tsx
index 233de6d..146aeba 100644
--- a/packages/react/stories/Dialog.stories.tsx
+++ b/packages/react/stories/Dialog.stories.tsx
@@ -85,15 +85,17 @@ export const WithCloseOnClick = () => {
             ad minim veniam, quis nostrud exercitation ullamco laboris nisi ut
             aliquip ex ea commodo consequat.
           </p>
-          <DialogClose>
-            <Button
-              onClick={(e) => {
-                console.log(`Closed dialog: `, e.currentTarget);
-              }}
-            >
-              Close Dialog
-            </Button>
-          </DialogClose>
+          <DialogFooter>
+            <DialogClose>
+              <Button
+                onClick={(e) => {
+                  console.log(`Closed dialog: `, e.currentTarget);
+                }}
+              >
+                Close Dialog
+              </Button>
+            </DialogClose>
+          </DialogFooter>
         </DialogContent>
       </DialogOverlay>
     </DialogRoot>