From c93b8ef7fce5af09f27b036cb61e24d86775fa06 Mon Sep 17 00:00:00 2001
From: Sebastian Malton <sebastian@malton.name>
Date: Mon, 2 May 2022 11:09:13 -0400
Subject: [PATCH] better fix

Signed-off-by: Sebastian Malton <sebastian@malton.name>
---
 src/renderer/components/confirm-dialog/confirm-dialog.tsx       | 2 +-
 .../components/kube-object-menu/kube-object-menu.test.tsx       | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

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", () => {