diff --git a/d2renderers/d2svg/d2svg.go b/d2renderers/d2svg/d2svg.go
index 9f5dddbe2..1d9855e9a 100644
--- a/d2renderers/d2svg/d2svg.go
+++ b/d2renderers/d2svg/d2svg.go
@@ -74,6 +74,7 @@ func setViewbox(writer io.Writer, diagram *d2target.Diagram, pad int) (width int
 	// TODO background stuff. e.g. dotted, grid, colors
 	fmt.Fprintf(writer, `<?xml version="1.0" encoding="utf-8"?>
 <svg
+id="d2-svg"
 style="background: white;"
 xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink"
 width="%d" height="%d" viewBox="%d %d %d %d">`, w, h, tl.X-pad, tl.Y-pad, w, h)
diff --git a/d2renderers/d2svg/fitToScreen.js b/d2renderers/d2svg/fitToScreen.js
index d753cf163..51e6d6ea4 100644
--- a/d2renderers/d2svg/fitToScreen.js
+++ b/d2renderers/d2svg/fitToScreen.js
@@ -1,10 +1,8 @@
 window.addEventListener("DOMContentLoaded", () => {
-  const svgEl = document.querySelector("svg");
-  const svgID = "d2-svg";
-  svgEl.setAttribute("id", svgID);
-  if (document.documentElement.getAttribute("id") !== svgID) {
+  if (document.documentElement.getAttribute("id") !== "d2-svg") {
     return;
   }
+  const svgEl = document.documentElement;
   let width = parseInt(svgEl.getAttribute("width"), 10);
   let height = parseInt(svgEl.getAttribute("height"), 10);
   let ratio;