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