diff --git a/src/renderer/components/confirm-dialog/confirm-dialog.tsx b/src/renderer/components/confirm-dialog/confirm-dialog.tsx
index 5bfe3dad60..5ce655de54 100644
--- a/src/renderer/components/confirm-dialog/confirm-dialog.tsx
+++ b/src/renderer/components/confirm-dialog/confirm-dialog.tsx
@@ -117,7 +117,7 @@ class NonInjectedConfirmDialog extends React.Component<ConfirmDialogProps & Depe
         isOpen={isOpen}
         onClose={this.onClose}
         close={this.close}
-        data-testid="confirmation-dialog"
+        {...(isOpen ? { "data-testid": "confirmation-dialog" } : {})}
       >
         <div className="confirm-content">
           {icon} {message}
diff --git a/src/renderer/components/kube-object-menu/kube-object-menu.test.tsx b/src/renderer/components/kube-object-menu/kube-object-menu.test.tsx
index b2ca15ae89..a73553c5cb 100644
--- a/src/renderer/components/kube-object-menu/kube-object-menu.test.tsx
+++ b/src/renderer/components/kube-object-menu/kube-object-menu.test.tsx
@@ -170,7 +170,7 @@ describe("kube-object-menu", () => {
 
       it("renders", async () => {
         await screen.findByTestId("confirmation-dialog");
-        await waitFor(() => expect(baseElement).toMatchSnapshot());
+        expect(baseElement).toMatchSnapshot();
       });
 
       describe("when remove is confirmed", () => {