From 9e1f488f22ebcb1912124461feb965c72fdef55b Mon Sep 17 00:00:00 2001
From: Wojtek Figat <wojtek@figat.pl>
Date: Sat, 13 Sep 2025 14:41:11 +0200
Subject: [PATCH] Fix prefab window to use UI Mode for empty UI Control prefab

#3574
---
 Source/Editor/Viewport/Previews/PrefabPreview.cs | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/Source/Editor/Viewport/Previews/PrefabPreview.cs b/Source/Editor/Viewport/Previews/PrefabPreview.cs
index 4ee763540..70316e4d9 100644
--- a/Source/Editor/Viewport/Previews/PrefabPreview.cs
+++ b/Source/Editor/Viewport/Previews/PrefabPreview.cs
@@ -112,8 +112,9 @@ namespace FlaxEditor.Viewport.Previews
             LinkCanvas(_instance);
 
             // Link UI control to the preview
+            var uiControl = _instance as UIControl;
             if (_uiControlLinked == null &&
-                _instance is UIControl uiControl &&
+                uiControl != null &&
                 uiControl.Control != null &&
                 uiControl.Control.Parent == null)
             {
@@ -128,6 +129,12 @@ namespace FlaxEditor.Viewport.Previews
                     _uiControlLinked.Control.Parent = _uiParentLink;
                 _hasUILinked = true;
             }
+
+            // Use UI mode when root is empty UI Control
+            if (_uiControlLinked == null && uiControl != null && uiControl.Control == null)
+            {
+                _hasUILinked = true;
+            }
         }
 
         private void LinkCanvas(Actor actor)