diff --git a/src/components/thread/messages/ai.tsx b/src/components/thread/messages/ai.tsx
index 80adf43..61c47c3 100644
--- a/src/components/thread/messages/ai.tsx
+++ b/src/components/thread/messages/ai.tsx
@@ -9,7 +9,7 @@ import { cn } from "@/lib/utils";
import { ToolCalls, ToolResult } from "./tool-calls";
import { MessageContentComplex } from "@langchain/core/messages";
import { Fragment } from "react/jsx-runtime";
-import { isAgentInboxInterrupt } from "@/lib/is-hitl";
+import { isAgentInboxInterruptSchema } from "@/lib/agent-inbox-interrupt";
import { ThreadView } from "../agent-inbox";
function CustomComponent({
@@ -118,7 +118,7 @@ export function AssistantMessage({
)) ||
(hasToolCalls &&