From 12e2b8156f5d78c797cf02d104102d055d1d0bc5 Mon Sep 17 00:00:00 2001
From: Wojtek Figat <wojtek@figat.pl>
Date: Mon, 18 Mar 2024 20:31:17 +0100
Subject: [PATCH] Fix UI prefab focus on canvas not being added as root

---
 Source/Editor/Viewport/PrefabWindowViewport.cs | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/Source/Editor/Viewport/PrefabWindowViewport.cs b/Source/Editor/Viewport/PrefabWindowViewport.cs
index 1bf121ce2..252981555 100644
--- a/Source/Editor/Viewport/PrefabWindowViewport.cs
+++ b/Source/Editor/Viewport/PrefabWindowViewport.cs
@@ -2,6 +2,7 @@
 
 using System;
 using System.Collections.Generic;
+using System.Linq;
 using FlaxEditor.Content;
 using FlaxEditor.Gizmo;
 using FlaxEditor.GUI.ContextMenu;
@@ -285,7 +286,11 @@ namespace FlaxEditor.Viewport
                 ShowDebugDraw = false;
 
                 // Show whole UI on startup
-                ViewportCamera.ShowActor(Instance);
+                var canvas = (CanvasRootControl)_uiParentLink.Children.FirstOrDefault(x => x is CanvasRootControl);
+                if (canvas != null)
+                    ViewportCamera.ShowActor(canvas.Canvas);
+                else if (Instance is UIControl)
+                    ViewportCamera.ShowActor(Instance);
             }
             else
             {