diff --git a/site/client.tsx b/site/highlight-code.tsx
similarity index 71%
rename from site/client.tsx
rename to site/highlight-code.tsx
index 3dc4726..7e83bfb 100644
--- a/site/client.tsx
+++ b/site/highlight-code.tsx
@@ -6,10 +6,6 @@ const highlighter = await createHighlighter({
 });
 
 for (const sample of document.querySelectorAll<HTMLElement>('.sample')) {
-  import(`./samples/${sample.dataset['sample']}.js`).then(mod => {
-    sample.querySelector('.sample-output')!.append(<mod.default />);
-  });
-
   const pre = sample.querySelector('pre')!;
   pre.outerHTML = highlighter.codeToHtml(pre.textContent!, {
     lang: 'tsx',
diff --git a/site/index.html.tsx b/site/index.html.tsx
index 704115c..58f74e3 100644
--- a/site/index.html.tsx
+++ b/site/index.html.tsx
@@ -7,7 +7,8 @@ export default <>
       <meta charset="UTF-8" />
       <meta name="viewport" content="width=device-width, initial-scale=1.0" />
       <link rel='stylesheet' href='style.css' />
-      <script type='module' src='client.js' />
+      <script type='module' src='load-samples.js' />
+      <script type='module' src='highlight-code.js' />
       <title>Vanilla JSX</title>
     </head>
     <body>
diff --git a/site/load-samples.tsx b/site/load-samples.tsx
new file mode 100644
index 0000000..63814a1
--- /dev/null
+++ b/site/load-samples.tsx
@@ -0,0 +1,5 @@
+for (const sample of document.querySelectorAll<HTMLElement>('.sample')) {
+  import(`./samples/${sample.dataset['sample']}.js`).then(mod => {
+    sample.querySelector('.sample-output')!.append(<mod.default />);
+  });
+}
