diff --git a/test/short-urls/helpers/DeleteShortUrlModal.test.tsx b/test/short-urls/helpers/DeleteShortUrlModal.test.tsx
index d4145296..448705ef 100644
--- a/test/short-urls/helpers/DeleteShortUrlModal.test.tsx
+++ b/test/short-urls/helpers/DeleteShortUrlModal.test.tsx
@@ -5,6 +5,7 @@ import { ShortUrl } from '../../../src/short-urls/data';
 import { ShortUrlDeletion } from '../../../src/short-urls/reducers/shortUrlDeletion';
 import { renderWithEvents } from '../../__helpers__/setUpTest';
 import { ErrorTypeV2, ErrorTypeV3, InvalidShortUrlDeletion, ProblemDetailsError } from '../../../src/api/types/errors';
+import { TestModalWrapper } from '../../__helpers__/TestModalWrapper';
 
 describe('<DeleteShortUrlModal />', () => {
   const shortUrl = Mock.of<ShortUrl>({
@@ -13,16 +14,19 @@ describe('<DeleteShortUrlModal />', () => {
     longUrl: 'https://long-domain.com/foo/bar',
   });
   const deleteShortUrl = jest.fn().mockResolvedValue(undefined);
-  const toggle = jest.fn();
+  const shortUrlDeleted = jest.fn();
   const setUp = (shortUrlDeletion: Partial<ShortUrlDeletion>) => renderWithEvents(
-    <DeleteShortUrlModal
-      isOpen
-      shortUrl={shortUrl}
-      shortUrlDeletion={Mock.of<ShortUrlDeletion>(shortUrlDeletion)}
-      deleteShortUrl={deleteShortUrl}
-      shortUrlDeleted={jest.fn()}
-      toggle={toggle}
-      resetDeleteShortUrl={jest.fn()}
+    <TestModalWrapper
+      renderModal={(args) => (
+        <DeleteShortUrlModal
+          {...args}
+          shortUrl={shortUrl}
+          shortUrlDeletion={Mock.of<ShortUrlDeletion>(shortUrlDeletion)}
+          deleteShortUrl={deleteShortUrl}
+          shortUrlDeleted={shortUrlDeleted}
+          resetDeleteShortUrl={jest.fn()}
+        />
+      )}
     />,
   );
 
@@ -76,6 +80,7 @@ describe('<DeleteShortUrlModal />', () => {
     const { user } = setUp({
       loading: false,
       error: false,
+      deleted: true,
       shortCode,
     });
 
@@ -83,6 +88,6 @@ describe('<DeleteShortUrlModal />', () => {
     await user.type(screen.getByPlaceholderText(/^Insert the short code/), shortCode);
     await user.click(screen.getByRole('button', { name: 'Delete' }));
     expect(deleteShortUrl).toHaveBeenCalledTimes(1);
-    await waitFor(() => expect(toggle).toHaveBeenCalledTimes(1));
+    await waitFor(() => expect(shortUrlDeleted).toHaveBeenCalledTimes(1));
   });
 });