diff --git a/frontend/src/Editor/Components/Table/AddNewRowComponent.jsx b/frontend/src/Editor/Components/Table/AddNewRowComponent.jsx
index 2f24dbac2d..9e353afc3e 100644
--- a/frontend/src/Editor/Components/Table/AddNewRowComponent.jsx
+++ b/frontend/src/Editor/Components/Table/AddNewRowComponent.jsx
@@ -161,11 +161,10 @@ export function AddNewRowComponent({
         <ButtonSolid
           variant="primary"
           className={`tj-text-xsm`}
-          onClick={() => {
-            onEvent('onNewRowsAdded', tableEvents, { component }).then(() => {
-              mergeToAddNewRowsDetails({ newRowsDataUpdates: {}, newRowsChangeSet: {}, addingNewRows: false });
-              setNewRowsState([]);
-            });
+          onClick={async () => {
+            await onEvent('onNewRowsAdded', tableEvents, { component });
+            mergeToAddNewRowsDetails({ newRowsDataUpdates: {}, newRowsChangeSet: {}, addingNewRows: false });
+            setNewRowsState([]);
           }}
           size="sm"
           customStyles={{ padding: '10px 20px' }}
diff --git a/frontend/src/Editor/EditorFunc.jsx b/frontend/src/Editor/EditorFunc.jsx
index 04e07591b0..5dfc167fa7 100644
--- a/frontend/src/Editor/EditorFunc.jsx
+++ b/frontend/src/Editor/EditorFunc.jsx
@@ -1041,9 +1041,7 @@ const EditorComponent = (props) => {
         undoOpts = {
           componentAdded: true,
         };
-      }
-
-      if (undoOpts?.componentAdded) {
+      } else if (undoOpts?.componentAdded) {
         undoOpts = {
           componentDeleted: true,
         };