diff --git a/lib/components/EditModal.tsx b/lib/components/EditModal.tsx
index 9ee3df7..03efd55 100644
--- a/lib/components/EditModal.tsx
+++ b/lib/components/EditModal.tsx
@@ -46,7 +46,9 @@ const CustomModal: React.FC<EditModalProps> = ({ title, visible, onClose, onSave
 
     const handleSave = () => {
         onSave( inputValues );
-        if (onClose) onClose();
+        if (onClose) {
+            onClose();
+        }
     };
 
     return (