diff --git a/src/hooks/claude-code-hooks/index.ts b/src/hooks/claude-code-hooks/index.ts
index 9f84639..acfc3ee 100644
--- a/src/hooks/claude-code-hooks/index.ts
+++ b/src/hooks/claude-code-hooks/index.ts
@@ -138,7 +138,7 @@ export function createClaudeCodeHooksHook(ctx: PluginInput, config: PluginConfig
           return
         }
 
-        const detectedKeywords = detectKeywordsWithType(removeCodeBlocks(prompt))
+        const detectedKeywords = detectKeywordsWithType(removeCodeBlocks(prompt), input.agent)
         const keywordMessages = detectedKeywords.map((k) => k.message)
 
         if (keywordMessages.length > 0) {
diff --git a/src/hooks/keyword-detector/index.ts b/src/hooks/keyword-detector/index.ts
index 13a4c22..462fd39 100644
--- a/src/hooks/keyword-detector/index.ts
+++ b/src/hooks/keyword-detector/index.ts
@@ -21,7 +21,7 @@ export function createKeywordDetectorHook(ctx: PluginInput) {
       }
     ): Promise<void> => {
       const promptText = extractPromptText(output.parts)
-      const detectedKeywords = detectKeywordsWithType(removeCodeBlocks(promptText))
+      const detectedKeywords = detectKeywordsWithType(removeCodeBlocks(promptText), input.agent)
 
       if (detectedKeywords.length === 0) {
         return