From 7b837ac8ee3e38c78838e347f89ea4642d70e138 Mon Sep 17 00:00:00 2001
From: Vedang <vedang685@gmail.com>
Date: Wed, 2 Oct 2024 19:26:29 +0530
Subject: [PATCH] [FIX] Passed all test cases

---
 packages/excalidraw/components/App.tsx | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/packages/excalidraw/components/App.tsx b/packages/excalidraw/components/App.tsx
index cc23b91f9..0b227afd4 100644
--- a/packages/excalidraw/components/App.tsx
+++ b/packages/excalidraw/components/App.tsx
@@ -1950,6 +1950,11 @@ class App extends React.Component<AppProps, AppState> {
         frameElement,
         data: { status: "done", html: parsedHtml },
       });
+
+      this.setState({
+        selectedElementIds: { [frameElement.id]: true },
+      });
+
     } catch (error: any) {
       trackEvent("ai", "generate (failed)", "d2c");
       this.updateMagicGeneration({
@@ -2991,8 +2996,6 @@ class App extends React.Component<AppProps, AppState> {
           ),
         });
 
-        this.actionManager.executeAction(actionFinalize);
-
         return;
       }