diff --git a/SmartSQL/SmartSQL/UserControl/GenCodes/UcGenCode.xaml.cs b/SmartSQL/SmartSQL/UserControl/GenCodes/UcGenCode.xaml.cs
index d69c601b3d229dce92d4d312b1d52934529f41a7..c04001beef9efca619fe83523c6ea54e1d192d04 100644
--- a/SmartSQL/SmartSQL/UserControl/GenCodes/UcGenCode.xaml.cs
+++ b/SmartSQL/SmartSQL/UserControl/GenCodes/UcGenCode.xaml.cs
@@ -795,7 +795,7 @@ namespace SmartSQL.UserControl.GenCodes
                 selectedConnectionString, selectedDatabase.DbName);
             foreach (var group in treeViewData)
             {
-                if (group.Type == "Type" && group.Name.Equals("treeTable"))
+                if (group.Type == "Type" && (group.Name.Equals("treeTable") || group.Name.Equals("treeView")))
                 {
                     int orderNo = 1;
                     foreach (var node in group.Children)