diff --git a/apps/app/src/components/promptbox/banner/ThreadPromptContextBanner.test.tsx b/apps/app/src/components/promptbox/banner/ThreadPromptContextBanner.test.tsx
index 8489ede7f4..4c0a0dafa2 100644
--- a/apps/app/src/components/promptbox/banner/ThreadPromptContextBanner.test.tsx
+++ b/apps/app/src/components/promptbox/banner/ThreadPromptContextBanner.test.tsx
@@ -150,6 +150,54 @@ describe("ThreadPromptContextBanner", () => {
},
);
+ it("renders an enabled Restore action once the environment is destroyed", () => {
+ const markup = renderToStaticMarkup(
+ ,
+ );
+
+ expect(markup).toContain("Restore environment");
+ expect(markup).toContain("