diff --git a/lib/folk-calendar.ts b/lib/folk-calendar.ts
index 329502a..6759b22 100644
--- a/lib/folk-calendar.ts
+++ b/lib/folk-calendar.ts
@@ -270,7 +270,8 @@ export class FolkCalendar extends FolkShape {
 		`;
 
 		// Replace the container div (slot's parent) with our wrapper
-		const containerDiv = root.querySelector(":scope > div");
+		const slot = root.querySelector("slot");
+		const containerDiv = slot?.parentElement as HTMLElement;
 		if (containerDiv) {
 			containerDiv.replaceWith(wrapper);
 		}
diff --git a/lib/folk-chat.ts b/lib/folk-chat.ts
index 7aa0139..7aff580 100644
--- a/lib/folk-chat.ts
+++ b/lib/folk-chat.ts
@@ -239,7 +239,8 @@ export class FolkChat extends FolkShape {
 		`;
 
 		// Replace the container div (slot's parent) with our wrapper
-		const containerDiv = root.querySelector(":scope > div");
+		const slot = root.querySelector("slot");
+		const containerDiv = slot?.parentElement as HTMLElement;
 		if (containerDiv) {
 			containerDiv.replaceWith(wrapper);
 		}
diff --git a/lib/folk-embed.ts b/lib/folk-embed.ts
index 90d19dd..77224a2 100644
--- a/lib/folk-embed.ts
+++ b/lib/folk-embed.ts
@@ -259,7 +259,8 @@ export class FolkEmbed extends FolkShape {
 		`;
 
 		// Replace the container div (slot's parent) with our wrapper
-		const containerDiv = root.querySelector(":scope > div");
+		const slot = root.querySelector("slot");
+		const containerDiv = slot?.parentElement as HTMLElement;
 		if (containerDiv) {
 			containerDiv.replaceWith(wrapper);
 		}
diff --git a/lib/folk-google-item.ts b/lib/folk-google-item.ts
index 27e5565..2f809c2 100644
--- a/lib/folk-google-item.ts
+++ b/lib/folk-google-item.ts
@@ -230,7 +230,8 @@ export class FolkGoogleItem extends FolkShape {
 		`;
 
 		// Replace the container div (slot's parent) with our card
-		const containerDiv = root.querySelector(":scope > div");
+		const slot = root.querySelector("slot");
+		const containerDiv = slot?.parentElement as HTMLElement;
 		const cardEl = wrapper.querySelector(".card");
 		if (containerDiv && cardEl) {
 			containerDiv.replaceWith(cardEl);
diff --git a/lib/folk-image-gen.ts b/lib/folk-image-gen.ts
index d28f076..45c5218 100644
--- a/lib/folk-image-gen.ts
+++ b/lib/folk-image-gen.ts
@@ -265,7 +265,8 @@ export class FolkImageGen extends FolkShape {
 		`;
 
 		// Replace the container div (slot's parent) with our wrapper
-		const containerDiv = root.querySelector(":scope > div");
+		const slot = root.querySelector("slot");
+		const containerDiv = slot?.parentElement as HTMLElement;
 		if (containerDiv) {
 			containerDiv.replaceWith(wrapper);
 		}
diff --git a/lib/folk-map.ts b/lib/folk-map.ts
index 6ff1673..d83336a 100644
--- a/lib/folk-map.ts
+++ b/lib/folk-map.ts
@@ -309,7 +309,8 @@ export class FolkMap extends FolkShape {
 		`;
 
 		// Replace the container div (slot's parent) with our wrapper
-		const containerDiv = root.querySelector(":scope > div");
+		const slot = root.querySelector("slot");
+		const containerDiv = slot?.parentElement as HTMLElement;
 		if (containerDiv) {
 			containerDiv.replaceWith(wrapper);
 		}
diff --git a/lib/folk-markdown.ts b/lib/folk-markdown.ts
index 56f9cc0..9bdec1c 100644
--- a/lib/folk-markdown.ts
+++ b/lib/folk-markdown.ts
@@ -174,7 +174,8 @@ export class FolkMarkdown extends FolkShape {
 		`;
 
 		// Replace the container div (slot's parent) with our wrapper
-		const containerDiv = root.querySelector(":scope > div");
+		const slot = root.querySelector("slot");
+		const containerDiv = slot?.parentElement as HTMLElement;
 		if (containerDiv) {
 			containerDiv.replaceWith(wrapper);
 		}
diff --git a/lib/folk-obs-note.ts b/lib/folk-obs-note.ts
index 30c1551..5582269 100644
--- a/lib/folk-obs-note.ts
+++ b/lib/folk-obs-note.ts
@@ -367,7 +367,8 @@ export class FolkObsNote extends FolkShape {
 		`;
 
 		// Replace the container div (slot's parent) with our wrapper
-		const containerDiv = root.querySelector(":scope > div");
+		const slot = root.querySelector("slot");
+		const containerDiv = slot?.parentElement as HTMLElement;
 		if (containerDiv) {
 			containerDiv.replaceWith(wrapper);
 		}
diff --git a/lib/folk-piano.ts b/lib/folk-piano.ts
index 9ca814d..095c572 100644
--- a/lib/folk-piano.ts
+++ b/lib/folk-piano.ts
@@ -196,7 +196,8 @@ export class FolkPiano extends FolkShape {
 		`;
 
 		// Replace the container div (slot's parent) with our piano container
-		const containerDiv = root.querySelector(":scope > div");
+		const slot = root.querySelector("slot");
+		const containerDiv = slot?.parentElement as HTMLElement;
 		const pianoContainer = wrapper.querySelector(".piano-container");
 		if (containerDiv && pianoContainer) {
 			containerDiv.replaceWith(pianoContainer);
diff --git a/lib/folk-prompt.ts b/lib/folk-prompt.ts
index 4249c29..d692f0b 100644
--- a/lib/folk-prompt.ts
+++ b/lib/folk-prompt.ts
@@ -294,7 +294,8 @@ export class FolkPrompt extends FolkShape {
 		`;
 
 		// Replace the container div (slot's parent) with our wrapper
-		const containerDiv = root.querySelector(":scope > div");
+		const slot = root.querySelector("slot");
+		const containerDiv = slot?.parentElement as HTMLElement;
 		if (containerDiv) {
 			containerDiv.replaceWith(wrapper);
 		}
diff --git a/lib/folk-slide.ts b/lib/folk-slide.ts
index a41a0cd..ff3d235 100644
--- a/lib/folk-slide.ts
+++ b/lib/folk-slide.ts
@@ -89,7 +89,8 @@ export class FolkSlide extends FolkShape {
 		`;
 
 		// Replace the container div (slot's parent) with our slide container
-		const containerDiv = root.querySelector(":scope > div");
+		const slot = root.querySelector("slot");
+		const containerDiv = slot?.parentElement as HTMLElement;
 		const slideContainer = wrapper.querySelector(".slide-container");
 		if (containerDiv && slideContainer) {
 			containerDiv.replaceWith(slideContainer);
diff --git a/lib/folk-transcription.ts b/lib/folk-transcription.ts
index 51c797b..078546a 100644
--- a/lib/folk-transcription.ts
+++ b/lib/folk-transcription.ts
@@ -350,7 +350,8 @@ export class FolkTranscription extends FolkShape {
 		`;
 
 		// Replace the container div (slot's parent) with our wrapper
-		const containerDiv = root.querySelector(":scope > div");
+		const slot = root.querySelector("slot");
+		const containerDiv = slot?.parentElement as HTMLElement;
 		if (containerDiv) {
 			containerDiv.replaceWith(wrapper);
 		}
diff --git a/lib/folk-video-chat.ts b/lib/folk-video-chat.ts
index c38903c..4a618d6 100644
--- a/lib/folk-video-chat.ts
+++ b/lib/folk-video-chat.ts
@@ -326,7 +326,8 @@ export class FolkVideoChat extends FolkShape {
 		`;
 
 		// Replace the container div (slot's parent) with our wrapper
-		const containerDiv = root.querySelector(":scope > div");
+		const slot = root.querySelector("slot");
+		const containerDiv = slot?.parentElement as HTMLElement;
 		if (containerDiv) {
 			containerDiv.replaceWith(wrapper);
 		}
diff --git a/lib/folk-video-gen.ts b/lib/folk-video-gen.ts
index d493a33..a644e2b 100644
--- a/lib/folk-video-gen.ts
+++ b/lib/folk-video-gen.ts
@@ -360,7 +360,8 @@ export class FolkVideoGen extends FolkShape {
 		`;
 
 		// Replace the container div (slot's parent) with our wrapper
-		const containerDiv = root.querySelector(":scope > div");
+		const slot = root.querySelector("slot");
+		const containerDiv = slot?.parentElement as HTMLElement;
 		if (containerDiv) {
 			containerDiv.replaceWith(wrapper);
 		}
diff --git a/lib/folk-workflow-block.ts b/lib/folk-workflow-block.ts
index 00e2bae..f1d473d 100644
--- a/lib/folk-workflow-block.ts
+++ b/lib/folk-workflow-block.ts
@@ -341,7 +341,8 @@ export class FolkWorkflowBlock extends FolkShape {
 		`;
 
 		// Replace the container div (slot's parent) with our wrapper
-		const containerDiv = root.querySelector(":scope > div");
+		const slot = root.querySelector("slot");
+		const containerDiv = slot?.parentElement as HTMLElement;
 		if (containerDiv) {
 			containerDiv.replaceWith(wrapper);
 		}