diff --git a/scripts/mock.js b/scripts/mock.js
index 69fcfa93..1781d331 100644
--- a/scripts/mock.js
+++ b/scripts/mock.js
@@ -380,11 +380,11 @@ exports.mock = async page => {
       json: []
     },
     shardShards: {
-      url: 'http://localhost:9200/_cat/shards/?h=index%2Cshard%2Cprirep%2Cstate%2Cnode',
+      url: 'http://localhost:9200/_cat/shards/?h=index%2Cshard%2Cprirep%2Cstate%2Cnode%2Cdocs%2Cstore%2Cip%2Cnode%2Cunassigned.reason',
       json: []
     },
     indexIndices: {
-      url: 'http://localhost:9200/_cat/indices/?h=index%2Chealth%2Cstatus%2Cuuid%2Cpri%2Crep%2Cdocs.count%2Cstore.size%2Csc&bytes=b',
+      url: 'http://localhost:9200/_cat/indices/?h=index%2Chealth%2Cstatus%2Cuuid%2Cpri%2Crep%2Cdocs.count%2Cstore.size%2Csc%2Ccd&bytes=b',
       json: []
     },
     flush: {
diff --git a/scripts/website_screenshots.js b/scripts/website_screenshots.js
index d788dc2c..455deb99 100644
--- a/scripts/website_screenshots.js
+++ b/scripts/website_screenshots.js
@@ -41,6 +41,7 @@ const screenshotIndexAliases = async page => {
 }
 const screenshotSearch = async page => (await navAndScreenshot(page, '#search', 'search', true))
 const screenshotEditDocument = async page => {
+  await page.locator('body').dblclick()
   await navAndScreenshot(page, 'body', 'document')
   await page.locator('#close').click()
 }
@@ -52,7 +53,8 @@ const screenshotRest = async page => {
 const screenshotSettings = async page => (await navAndScreenshot(page, '#settings', 'settings'))
 
 const toggleTheme = async page => {
-  await page.locator('#change_theme').click()
+  await page.locator('[data-testid="change-theme-button"]').click()
+  await page.locator('[data-testid="change-theme__dark"]').click()
 }
 
 const connectWithServer = async page => {
@@ -66,10 +68,6 @@ const navAndScreenshot = async (page, selector, name, persisted) => {
   await screenshot(page, name, persisted)
 }
 
-const closeSnackbar = async page => {
-  await page.locator('#close_snackbar').click()
-}
-
 let counter = 1
 const screenshot = async (page, name, persisted) => {
   if (!persisted) await page.locator('body').click()