From da26a1947f9ca2b300c9078e4e60417f2dd38870 Mon Sep 17 00:00:00 2001
From: KaustubhKumar05 <kaustubhkumar05@gmail.com>
Date: Wed, 11 Oct 2023 11:00:00 +0530
Subject: [PATCH] fix: types

---
 packages/roomkit-react/src/Modal/Dialog.tsx | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/packages/roomkit-react/src/Modal/Dialog.tsx b/packages/roomkit-react/src/Modal/Dialog.tsx
index f64e7b977b..12630c7a91 100644
--- a/packages/roomkit-react/src/Modal/Dialog.tsx
+++ b/packages/roomkit-react/src/Modal/Dialog.tsx
@@ -23,11 +23,17 @@ const CustomDialogContent = ({ children, props = {}, css = {} }: { children: Rea
 const CustomDialogOverlay = ({ css = {} }: { css?: CSS }) => (
   <StyledDialogOverlay css={{ ...css, position: 'absolute' }} />
 );
-const CustomDialogPortal = ({ children }: { children: ReactNode }) => {
+const CustomDialogPortal = ({ children, container }: { children: ReactNode; container: HTMLElement | null }) => {
   const dialogContainerSelector = useDialogContainerSelector();
   return (
     <StyledDialogPortal
-      container={dialogContainerSelector ? (document.querySelector(dialogContainerSelector) as HTMLElement) : undefined}
+      container={
+        container
+          ? container
+          : dialogContainerSelector
+          ? (document.querySelector(dialogContainerSelector) as HTMLElement)
+          : undefined
+      }
     >
       {children}
     </StyledDialogPortal>