diff --git a/apps/web/src/pages/titles/hooks/useTitleAI.ts b/apps/web/src/pages/titles/hooks/useTitleAI.ts
index ed0b19dad..3544fd735 100644
--- a/apps/web/src/pages/titles/hooks/useTitleAI.ts
+++ b/apps/web/src/pages/titles/hooks/useTitleAI.ts
@@ -1,11 +1,12 @@
 import { useState, useCallback } from "react"
 import { message } from "antd"
 import type { UseMutationResult } from "@tanstack/react-query"
+import type { TitleItem } from "@/api/titles"
 import { copyToClipboard } from "../utils/titleLibrary"
 import { AI_GENERATE_DELAY } from "../constants/titleLibrary"
 
 interface UseTitleAIProps {
-  createMutation: UseMutationResult<void, Error, string, unknown>
+  createMutation: UseMutationResult<TitleItem, Error, string, unknown>
 }
 
 const generateMockTitles = (keyword: string): string[] => [