From 9a39435c7d61ef6aec4d31d3ccbc7cfd1af670be Mon Sep 17 00:00:00 2001
From: velzie <velzie@velzie.rip>
Date: Sun, 25 Aug 2024 11:01:05 -0400
Subject: [PATCH] fix ownerDocument

---
 src/client/dom/element.ts | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/client/dom/element.ts b/src/client/dom/element.ts
index 4b25d04..b14f3a2 100644
--- a/src/client/dom/element.ts
+++ b/src/client/dom/element.ts
@@ -177,7 +177,13 @@ export default function (client: ScramjetClient, self: typeof window) {
 
 	client.Trap("Node.prototype.ownerDocument", {
 		get(ctx) {
-			return client.documentProxy;
+			let doc = ctx.get() as Document | null;
+			if (!doc) return null;
+
+			let scram: ScramjetClient = doc[ScramjetClient.SCRAMJET];
+			if (!scram) return doc; // ??
+
+			return scram.documentProxy;
 		},
 	});
 }