From 20e0de606bdd3e4dd59be54a651fd44c84375bb1 Mon Sep 17 00:00:00 2001
From: ALEXTANG <574809918@qq.com>
Date: Tue, 6 Sep 2022 19:12:57 +0800
Subject: [PATCH] Update ScriptGenerator.cs

---
 Assets/TEngine/Scripts/Editor/UI/ScriptGenerator.cs | 1 +
 1 file changed, 1 insertion(+)

diff --git a/Assets/TEngine/Scripts/Editor/UI/ScriptGenerator.cs b/Assets/TEngine/Scripts/Editor/UI/ScriptGenerator.cs
index 5eef71ef..47d49123 100644
--- a/Assets/TEngine/Scripts/Editor/UI/ScriptGenerator.cs
+++ b/Assets/TEngine/Scripts/Editor/UI/ScriptGenerator.cs
@@ -55,6 +55,7 @@ namespace TEngine.Editor
                     strFile.Append("using TEngine;\n");
                     strFile.Append("using UnityEngine;\n");
                     strFile.Append("using UnityEngine.UI;\n\n");
+                    strFile.Append("using TEngine.Runtime.UIModule;\n\n");
                     strFile.Append("\tclass " + root.name + " : UIWindow\n");
                     strFile.Append("\t{\n");
                 }