diff --git a/e2e/content-services/unshare_file.e2e.ts b/e2e/content-services/unshare_file.e2e.ts
index f2bb2d5e5b..89712a5cd1 100644
--- a/e2e/content-services/unshare_file.e2e.ts
+++ b/e2e/content-services/unshare_file.e2e.ts
@@ -90,10 +90,11 @@ describe('Unshare file', () => {
         let sharedLink = await shareDialog.getShareLink();
 
         shareDialog.clickShareToggle();
+        shareDialog.clickCloseButton();
 
         browser.get(sharedLink);
+        browser.sleep(300);
 
-        errorPage.checkErrorPage();
-        errorPage.checkBackButton();
+        errorPage.checkErrorCode();
     });
 });
diff --git a/e2e/pages/adf/dialog/shareDialog.ts b/e2e/pages/adf/dialog/shareDialog.ts
index 38166ebb71..60aa93dc11 100644
--- a/e2e/pages/adf/dialog/shareDialog.ts
+++ b/e2e/pages/adf/dialog/shareDialog.ts
@@ -23,6 +23,7 @@ export class ShareDialog {
     dialogTitle = element(by.id('adf-share-title'));
     shareToggle = element(by.id('adf-share-toggle'));
     shareLink = element(by.id('adf-share-link'));
+    closeButton = element(by.css('button[class="mat-button mat-primary"]'));
 
     checkDialogIsDisplayed() {
         return Util.waitUntilElementIsVisible(this.dialogTitle);
@@ -43,4 +44,9 @@ export class ShareDialog {
         Util.waitUntilElementIsVisible(this.shareLink)
         return this.shareLink.getAttribute('value');
     }
+
+    clickCloseButton() {
+        Util.waitUntilElementIsVisible(this.closeButton);
+        return this.closeButton.click();
+    }
 }