diff --git a/packages/core/client/src/schema-settings/GeneralSchemaDesigner.tsx b/packages/core/client/src/schema-settings/GeneralSchemaDesigner.tsx
index 7ba10563d5..cae1ab751b 100644
--- a/packages/core/client/src/schema-settings/GeneralSchemaDesigner.tsx
+++ b/packages/core/client/src/schema-settings/GeneralSchemaDesigner.tsx
@@ -310,8 +310,6 @@ const InternalSchemaToolbar: FC<SchemaToolbarProps> = React.memo((props) => {
   }, [schemaSettingsExists, schemaSettingsRender, settings]);
   const toolbarRef = useRef<HTMLDivElement>(null);
 
-  const hiddenClassName = process.env.__E2E__ ? 'hidden-e2e' : 'hidden';
-
   useEffect(() => {
     const toolbarElement = toolbarRef.current;
     let parentElement = toolbarElement?.parentElement;
@@ -327,14 +325,22 @@ const InternalSchemaToolbar: FC<SchemaToolbarProps> = React.memo((props) => {
 
     function show() {
       if (toolbarElement) {
-        toolbarElement.classList.remove(hiddenClassName);
+        if (process.env.__E2E__) {
+          toolbarElement.style.display = 'block';
+        } else {
+          toolbarElement.classList.remove('hidden');
+        }
         props.onVisibleChange?.(true);
       }
     }
 
     function hide() {
       if (toolbarElement) {
-        toolbarElement.classList.add(hiddenClassName);
+        if (process.env.__E2E__) {
+          toolbarElement.style.display = 'none';
+        } else {
+          toolbarElement.classList.add('hidden');
+        }
         props.onVisibleChange?.(false);
       }
     }
diff --git a/packages/core/client/src/schema-settings/styles.ts b/packages/core/client/src/schema-settings/styles.ts
index 4439f000db..786821afed 100644
--- a/packages/core/client/src/schema-settings/styles.ts
+++ b/packages/core/client/src/schema-settings/styles.ts
@@ -31,10 +31,6 @@ export const useStyles = genStyleHook('nb-schema-toolbar', (token) => {
         pointerEvents: 'none',
       },
 
-      '&.hidden-e2e': {
-        display: 'none',
-      },
-
       '.ant-space-item .anticon': {
         margin: 0,
       },