diff --git a/tests/McpContext.test.ts b/tests/McpContext.test.ts
index 57327117..63887a05 100644
--- a/tests/McpContext.test.ts
+++ b/tests/McpContext.test.ts
@@ -10,14 +10,19 @@ import sinon from 'sinon';
 
 import type {TraceResult} from '../src/trace-processing/parse.js';
 
-import {withBrowser} from './utils.js';
+import {html, withBrowser} from './utils.js';
 
 describe('McpContext', () => {
   it('list pages', async () => {
     await withBrowser(async (_response, context) => {
       const page = context.getSelectedPage();
-      await page.setContent(`<!DOCTYPE html>
-<button>Click me</button><input type="text" value="Input">`);
+      await page.setContent(
+        html`<button>Click me</button
+          ><input
+            type="text"
+            value="Input"
+          />`,
+      );
       await context.createTextSnapshot();
       assert.ok(await context.getElementByUid('1_1'));
       await context.createTextSnapshot();
diff --git a/tests/McpResponse.test.ts b/tests/McpResponse.test.ts
index 29dad971..1c6678d5 100644
--- a/tests/McpResponse.test.ts
+++ b/tests/McpResponse.test.ts
@@ -54,8 +54,13 @@ Testing 2`,
   it('returns correctly formatted snapshot for a simple tree', async () => {
     await withBrowser(async (response, context) => {
       const page = context.getSelectedPage();
-      await page.setContent(`<!DOCTYPE html>
-<button>Click me</button><input type="text" value="Input">`);
+      await page.setContent(
+        html`<button>Click me</button
+          ><input
+            type="text"
+            value="Input"
+          />`,
+      );
       await page.focus('button');
       response.includeSnapshot();
       const result = await response.handle('test', context);
@@ -64,7 +69,7 @@ Testing 2`,
         result[0].text,
         `# test response
 ## Page content
-uid=1_0 RootWebArea
+uid=1_0 RootWebArea "My test page"
   uid=1_1 button "Click me" focusable focused
   uid=1_2 textbox value="Input"
 `,
diff --git a/tests/tools/input.test.ts b/tests/tools/input.test.ts
index d819d80a..b6f39622 100644
--- a/tests/tools/input.test.ts
+++ b/tests/tools/input.test.ts
@@ -29,7 +29,7 @@ describe('input', () => {
       await withBrowser(async (response, context) => {
         const page = context.getSelectedPage();
         await page.setContent(
-          `<!DOCTYPE html><button onclick="this.innerText = 'clicked';">test`,
+          html`<button onclick="this.innerText = 'clicked';">test</button>`,
         );
         await context.createTextSnapshot();
         await click.handler(
@@ -53,7 +53,9 @@ describe('input', () => {
       await withBrowser(async (response, context) => {
         const page = context.getSelectedPage();
         await page.setContent(
-          `<!DOCTYPE html><button ondblclick="this.innerText = 'dblclicked';">test`,
+          html`<button ondblclick="this.innerText = 'dblclicked';"
+            >test</button
+          >`,
         );
         await context.createTextSnapshot();
         await click.handler(
@@ -158,7 +160,7 @@ describe('input', () => {
       await withBrowser(async (response, context) => {
         const page = context.getSelectedPage();
         await page.setContent(
-          `<!DOCTYPE html><button onmouseover="this.innerText = 'hovered';">test`,
+          html`<button onmouseover="this.innerText = 'hovered';">test</button>`,
         );
         await context.createTextSnapshot();
         await hover.handler(
@@ -184,7 +186,7 @@ describe('input', () => {
     it('fills out an input', async () => {
       await withBrowser(async (response, context) => {
         const page = context.getSelectedPage();
-        await page.setContent(`<!DOCTYPE html><input>`);
+        await page.setContent(html`<input />`);
         await context.createTextSnapshot();
         await fill.handler(
           {
@@ -209,7 +211,10 @@ describe('input', () => {
       await withBrowser(async (response, context) => {
         const page = context.getSelectedPage();
         await page.setContent(
-          `<!DOCTYPE html><select><option value="v1">one</option><option value="v2">two</option></select>`,
+          html`<select
+            ><option value="v1">one</option
+            ><option value="v2">two</option></select
+          >`,
         );
         await context.createTextSnapshot();
         await fill.handler(
@@ -239,25 +244,35 @@ describe('input', () => {
     it('drags one element onto another', async () => {
       await withBrowser(async (response, context) => {
         const page = context.getSelectedPage();
-        await page.setContent(`<!DOCTYPE html>
-<div role="button" id="drag" draggable="true">drag me</div>
-<div id="drop" aria-label="drop"
-  style="width: 100px; height: 100px; border: 1px solid black;" ondrop="this.innerText = 'dropped';">
-</div>
-<script>
-    drag.addEventListener("dragstart", (event) => {
-        event.dataTransfer.setData("text/plain", event.target.id);
-    });
-    drop.addEventListener("dragover", (event) => {
-        event.preventDefault();
-        event.dataTransfer.dropEffect = "move";
-    });
-    drop.addEventListener("drop", (event) => {
-        event.preventDefault();
-        const data = event.dataTransfer.getData("text/plain");
-        event.target.appendChild(document.getElementById(data));
-    });
-</script>`);
+        await page.setContent(
+          html`<div
+              role="button"
+              id="drag"
+              draggable="true"
+              >drag me</div
+            >
+            <div
+              id="drop"
+              aria-label="drop"
+              style="width: 100px; height: 100px; border: 1px solid black;"
+              ondrop="this.innerText = 'dropped';"
+            >
+            </div>
+            <script>
+              drag.addEventListener('dragstart', event => {
+                event.dataTransfer.setData('text/plain', event.target.id);
+              });
+              drop.addEventListener('dragover', event => {
+                event.preventDefault();
+                event.dataTransfer.dropEffect = 'move';
+              });
+              drop.addEventListener('drop', event => {
+                event.preventDefault();
+                const data = event.dataTransfer.getData('text/plain');
+                event.target.appendChild(document.getElementById(data));
+              });
+            </script>`,
+        );
         await context.createTextSnapshot();
         await drag.handler(
           {
@@ -283,12 +298,24 @@ describe('input', () => {
     it('successfully fills out the form', async () => {
       await withBrowser(async (response, context) => {
         const page = context.getSelectedPage();
-        await page.setContent(`<!DOCTYPE html>
-<form>
-  <label>username<input name=username type="text"/></label>
-  <label>email<input name=email type="text"/></label>
-  <input type=submit value="Submit">
-</form>`);
+        await page.setContent(
+          html`<form>
+            <label
+              >username<input
+                name="username"
+                type="text"
+            /></label>
+            <label
+              >email<input
+                name="email"
+                type="text"
+            /></label>
+            <input
+              type="submit"
+              value="Submit"
+            />
+          </form>`,
+        );
         await context.createTextSnapshot();
         await fillForm.handler(
           {
@@ -335,10 +362,14 @@ describe('input', () => {
 
       await withBrowser(async (response, context) => {
         const page = context.getSelectedPage();
-        await page.setContent(`<!DOCTYPE html>
-<form>
-  <input type="file" id="file-input">
-</form>`);
+        await page.setContent(
+          html`<form>
+            <input
+              type="file"
+              id="file-input"
+            />
+          </form>`,
+        );
         await context.createTextSnapshot();
         await uploadFile.handler(
           {
@@ -366,14 +397,21 @@ describe('input', () => {
 
       await withBrowser(async (response, context) => {
         const page = context.getSelectedPage();
-        await page.setContent(`<!DOCTYPE html>
-<button id="file-chooser-button">Upload file</button>
-<input type="file" id="file-input" style="display: none;">
-<script>
-  document.getElementById('file-chooser-button').addEventListener('click', () => {
-    document.getElementById('file-input').click();
-  });
-</script>`);
+        await page.setContent(
+          html`<button id="file-chooser-button">Upload file</button>
+            <input
+              type="file"
+              id="file-input"
+              style="display: none;"
+            />
+            <script>
+              document
+                .getElementById('file-chooser-button')
+                .addEventListener('click', () => {
+                  document.getElementById('file-input').click();
+                });
+            </script>`,
+        );
         await context.createTextSnapshot();
         await uploadFile.handler(
           {
@@ -406,7 +444,7 @@ describe('input', () => {
 
       await withBrowser(async (response, context) => {
         const page = context.getSelectedPage();
-        await page.setContent(`<!DOCTYPE html><div>Not a file input</div>`);
+        await page.setContent(html`<div>Not a file input</div>`);
         await context.createTextSnapshot();
 
         await assert.rejects(
@@ -465,11 +503,13 @@ describe('input', () => {
     it('processes press_key', async () => {
       await withBrowser(async (response, context) => {
         const page = context.getSelectedPage();
-        await page.setContent(`<!DOCTYPE html><script>
-logs=[];
-document.addEventListener('keydown', e => logs.push('d'+e.key));
-document.addEventListener('keyup', e => logs.push('u'+e.key));
-</script>`);
+        await page.setContent(
+          html`<script>
+            logs = [];
+            document.addEventListener('keydown', e => logs.push('d' + e.key));
+            document.addEventListener('keyup', e => logs.push('u' + e.key));
+          </script>`,
+        );
         await context.createTextSnapshot();
 
         await pressKey.handler(
