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; }, }); }