diff --git a/lib/db/schema.ts b/lib/db/schema.ts
index 202086c..882b061 100644
--- a/lib/db/schema.ts
+++ b/lib/db/schema.ts
@@ -10,6 +10,7 @@ import {
   foreignKey,
   boolean,
 } from 'drizzle-orm/pg-core';
+import { blockKinds } from '../blocks/server';
 
 export const user = pgTable('User', {
   id: uuid('id').primaryKey().notNull().defaultRandom(),
@@ -72,9 +73,7 @@ export const document = pgTable(
     createdAt: timestamp('createdAt').notNull(),
     title: text('title').notNull(),
     content: text('content'),
-    kind: varchar('text', { enum: ['text', 'code', 'image', 'sheet'] })
-      .notNull()
-      .default('text'),
+    kind: varchar('text', { enum: blockKinds }).notNull().default('text'),
     userId: uuid('userId')
       .notNull()
       .references(() => user.id),