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