diff --git a/src/Library/Infrastructure/Templates/Models/ClassBuildModel.cs b/src/Library/Infrastructure/Templates/Models/ClassBuildModel.cs
index 1971899..2f10252 100644
--- a/src/Library/Infrastructure/Templates/Models/ClassBuildModel.cs
+++ b/src/Library/Infrastructure/Templates/Models/ClassBuildModel.cs
@@ -39,10 +39,11 @@ public string PrimaryKeyTypeName
         {
             get
             {
-                var name = BaseEntityType.ToDescription();
-                if (name.Contains("Int")) return "int";
-                if (name.Contains("Long")) return "long";
-                return "Guid";
+                var name = BaseEntityType.ToDescription().ToLower();
+                if (name.Contains("int")) return "int";
+                if (name.Contains("long")) return "long";
+                if (name.Contains("guid")) return "Guid";
+                return "string";
             }
         }