diff --git a/Assets/TEngine/Editor/UI/ScriptGenerator.cs b/Assets/TEngine/Editor/UI/ScriptGenerator.cs
index dbe3f149..928e5e29 100644
--- a/Assets/TEngine/Editor/UI/ScriptGenerator.cs
+++ b/Assets/TEngine/Editor/UI/ScriptGenerator.cs
@@ -53,6 +53,7 @@ namespace TEngine.Editor
 
                 if (includeListener)
                 {
+                    strFile.Append("using UI;\n");
                     strFile.Append("using TEngine;\n");
                     strFile.Append("using UnityEngine;\n");
                     strFile.Append("using UnityEngine.UI;\n\n");