diff --git a/src/lib/databaseCapabilitySets.ts b/src/lib/databaseCapabilitySets.ts
index a985437248..5f16415b2f 100644
--- a/src/lib/databaseCapabilitySets.ts
+++ b/src/lib/databaseCapabilitySets.ts
@@ -114,6 +114,8 @@ export const TREE_SCHEMA_TYPES = new Set<DatabaseType>([
   "redshift",
   "sqlserver",
   "gaussdb",
+  "kingbase",
+  "vastbase",
   "jdbc",
   "trino",
   "h2",