diff --git a/packages/nc-gui/components/smartsheet/Gallery.vue b/packages/nc-gui/components/smartsheet/Gallery.vue
index 1ef419e23b..3018481b80 100644
--- a/packages/nc-gui/components/smartsheet/Gallery.vue
+++ b/packages/nc-gui/components/smartsheet/Gallery.vue
@@ -157,7 +157,7 @@ provide(ReloadRowDataHookInj, reloadViewDataHook)
 </script>
 
 <template>
-  <div class="flex flex-col h-full w-full overflow-auto">
+  <div class="flex flex-col h-full w-full overflow-auto nc-gallery">
     <div class="nc-gallery-container grid gap-2 my-4 px-3">
       <div v-for="record in data" :key="`record-${record.row.id}`">
         <Row :row="record">
diff --git a/scripts/cypress/integration/common/4g_table_view_expanded_form.js b/scripts/cypress/integration/common/4g_table_view_expanded_form.js
index a3a23514dc..2b73bf0670 100644
--- a/scripts/cypress/integration/common/4g_table_view_expanded_form.js
+++ b/scripts/cypress/integration/common/4g_table_view_expanded_form.js
@@ -89,7 +89,9 @@ export const genTest = (apiType, dbType) => {
         cy.url()
           .then((url) => {
             cy.visit('/' + url.split('/').slice(3).join('/').split('?')[0] + '?rowId=99999999');
+            cy.toastWait('Record not found');
             cy.get('.nc-expanded-form-header').should('not.exist');
+            cy.get(viewType === 'grid' ? '.nc-grid' : '.nc-gallery').should('exist');
           });
       });