From 59531b6dd2e8e068fbf781f53d9716d2cae35e98 Mon Sep 17 00:00:00 2001
From: Arturo Silva <arturo.silva@washpost.com>
Date: Thu, 6 Jun 2024 00:37:35 -0400
Subject: [PATCH] fix: also remove RTG from scrim too

---
 packages/kit/src/scrim/Scrim.tsx | 17 +++++++++++------
 1 file changed, 11 insertions(+), 6 deletions(-)

diff --git a/packages/kit/src/scrim/Scrim.tsx b/packages/kit/src/scrim/Scrim.tsx
index 8d49c46b8..29c9d8e91 100644
--- a/packages/kit/src/scrim/Scrim.tsx
+++ b/packages/kit/src/scrim/Scrim.tsx
@@ -1,4 +1,4 @@
-import React from "react";
+import React, { useTransition } from "react";
 import { styled, theme, css as wpCSS } from "../theme";
 import type * as WPDS from "../theme";
 import type { Token } from "@stitches/react/types/theme";
@@ -63,21 +63,26 @@ export const Scrim = React.forwardRef<HTMLDivElement, ScrimProps>(
       }
     }, [open, lockScroll]);
 
+    const [isPending, startTransition] = useTransition();
     const [isMounted, setIsMounted] = React.useState(false);
 
     React.useEffect(() => {
-      if (open) {
-        setIsMounted(true);
-      }
+      startTransition(() => {
+        if (open) {
+          setIsMounted(true);
+        }
+      });
     }, [open]);
 
     const handleTransitionEnd = () => {
       if (!open) {
-        setIsMounted(false);
+        startTransition(() => {
+          setIsMounted(false);
+        });
       }
     };
 
-    return isMounted ? (
+    return isMounted || isPending ? (
       <StyledContainer
         data-state={open ? "open" : "closed"}
         ref={ref}