diff --git a/docs/src/lib/examples.ts b/docs/src/lib/examples.ts
index a1a79428e..f886479a9 100644
--- a/docs/src/lib/examples.ts
+++ b/docs/src/lib/examples.ts
@@ -28,7 +28,7 @@ export async function loadExample(
 			default: Component;
 			layers?: string[];
 		};
-		const source = (rawSource.default as string).replace(/^.*export .*;.*$/gm, '');
+		const source = (rawSource.default as string).replace(/(\n\s*)*^.*export .*;.*$(\n\s*)*/gm, '\n');
 
 		return { component: comp, source, module };
 	} catch (e) {
@@ -84,7 +84,7 @@ export async function loadExampleByPath(resolvedPath: string): Promise<LoadedExa
 		}
 
 		const comp = componentModule.default as Component;
-		const source = (rawSource as string).replace(/^.*export .*;.*$/gm, '');
+		const source = (rawSource as string).replace(/(\n\s*)*^.*export .*;.*$(\n\s*)*/gm, '\n');
 
 		return { component: comp, source };
 	} catch (e) {
diff --git a/docs/src/lib/stackblitz.remote.ts b/docs/src/lib/stackblitz.remote.ts
index fb945babd..d048d6242 100644
--- a/docs/src/lib/stackblitz.remote.ts
+++ b/docs/src/lib/stackblitz.remote.ts
@@ -177,7 +177,7 @@ export const getExample = query(
 		}
 
 		// Remove `export { data };` from the code
-		const cleanCode = code.replace(/^.*export .*;.*$/gm, '');
+		const cleanCode = code.replace(/(\n\s*)*^.*export .*;.*$(\n\s*)*/gm, '\n');
 
 		// Prepare additional files to include
 		const additionalFiles: Record<string, string> = {};
diff --git a/docs/src/routes/docs/screenshot/[component]/+layout.ts b/docs/src/routes/docs/screenshot/[component]/+layout.ts
index 53b3564c3..e281ce910 100644
--- a/docs/src/routes/docs/screenshot/[component]/+layout.ts
+++ b/docs/src/routes/docs/screenshot/[component]/+layout.ts
@@ -18,7 +18,7 @@ export const load = async ({ params }) => {
 			const source = (await componentSources[path]()) as string;
 
 			// Remove `export { data };`
-			const cleanupSource = source.replace(/^.*export .*;.*$/gm, '');
+			const cleanupSource = source.replace(/(\n\s*)*^.*export .*;.*$(\n\s*)*/gm, '\n');
 
 			if (!examples[comp]) {
 				examples[comp] = {};