1 files changed, 2 insertions, 4 deletions

diff --git a/frontend/src/hooks/UseMessage.tsx b/frontend/src/hooks/UseMessage.tsx
index 9bfb713..13d8e4f 100644
--- a/frontend/src/hooks/UseMessage.tsx
+++ b/frontend/src/hooks/UseMessage.tsx
@@ -26,13 +26,11 @@ const useMessage = () => {
     };
 
     const MessageDialogComponent = isOpen && (
+                <div className="dialog-container">
         <MessageDialog title={title} subtitle={subtitle} onClose={handleClose}></MessageDialog>
+                </div>
     );
 
-        const getDialogComponent = () => {
-                return MessageDialogComponent;
-        };
-
     return { message, MessageDialogComponent };
 }