diff --git a/packages/frontend/core/src/blocksuite/ai/components/ai-tools/artifact-tool.ts b/packages/frontend/core/src/blocksuite/ai/components/ai-tools/artifact-tool.ts
index 03c571a07..684783e49 100644
--- a/packages/frontend/core/src/blocksuite/ai/components/ai-tools/artifact-tool.ts
+++ b/packages/frontend/core/src/blocksuite/ai/components/ai-tools/artifact-tool.ts
@@ -34,7 +34,7 @@ export abstract class ArtifactTool<
       padding: 10px 0;
 
       .affine-embed-linked-doc-block {
-        box-shadow: ${unsafeCSSVar('overlayPanelShadow')};
+        box-shadow: ${unsafeCSSVar('buttonShadow')};
         cursor: pointer;
       }
 
diff --git a/packages/frontend/core/src/blocksuite/ai/components/ai-tools/artifacts-preview-panel.ts b/packages/frontend/core/src/blocksuite/ai/components/ai-tools/artifacts-preview-panel.ts
index 34eb72e7e..a962b4ad0 100644
--- a/packages/frontend/core/src/blocksuite/ai/components/ai-tools/artifacts-preview-panel.ts
+++ b/packages/frontend/core/src/blocksuite/ai/components/ai-tools/artifacts-preview-panel.ts
@@ -104,10 +104,6 @@ export class ArtifactPreviewPanel extends WithDisposable(ShadowlessElement) {
       color: ${unsafeCSSVarV2('icon/secondary')};
     }
 
-    .artifact-panel-content {
-      height: calc(100% - 52px);
-    }
-
     .artifact-panel-close:hover {
       background-color: ${unsafeCSSVarV2('layer/background/tertiary')};
     }
diff --git a/packages/frontend/core/src/blocksuite/ai/components/ai-tools/doc-compose.ts b/packages/frontend/core/src/blocksuite/ai/components/ai-tools/doc-compose.ts
index 66794fa79..0f65b67ab 100644
--- a/packages/frontend/core/src/blocksuite/ai/components/ai-tools/doc-compose.ts
+++ b/packages/frontend/core/src/blocksuite/ai/components/ai-tools/doc-compose.ts
@@ -124,6 +124,7 @@ export class DocComposeTool extends ArtifactTool<
             .options=${{
               customHeading: true,
               extensions: getCustomPageEditorBlockSpecs(),
+              theme: this.theme,
             }}
           ></text-renderer>`
         : html`<div class="doc-compose-result-preview-loading">