diff --git a/keep-ui/app/(keep)/workflows/preview/[workflowId]/page.tsx b/keep-ui/app/(keep)/workflows/preview/[workflowId]/page.tsx
index b524b249f..6bfd54497 100644
--- a/keep-ui/app/(keep)/workflows/preview/[workflowId]/page.tsx
+++ b/keep-ui/app/(keep)/workflows/preview/[workflowId]/page.tsx
@@ -1,8 +1,8 @@
 "use client";
 import { useEffect, useState } from "react";
-import PageClient from "../../builder/page.client";
 import Loading from "@/app/(keep)/loading";
 import Link from "next/link";
+import { WorkflowBuilderPageClient } from "../../builder/page.client";
 
 export default function PageWithId({
   params,
@@ -27,8 +27,8 @@ export default function PageWithId({
     <>
       {!workflowPreviewData && <Loading />}
       {workflowPreviewData && workflowPreviewData.name === key && (
-        <PageClient
-          workflow={workflowPreviewData.workflow_raw || ""}
+        <WorkflowBuilderPageClient
+          workflowRaw={workflowPreviewData.workflow_raw || ""}
           isPreview={true}
         />
       )}
diff --git a/keep-ui/app/(keep)/workflows/preview/page.tsx b/keep-ui/app/(keep)/workflows/preview/page.tsx
index a4043b395..04be107fa 100644
--- a/keep-ui/app/(keep)/workflows/preview/page.tsx
+++ b/keep-ui/app/(keep)/workflows/preview/page.tsx
@@ -1,7 +1,7 @@
 "use client";
-import { useEffect, useState, Suspense } from "react";
+import { useEffect, useState } from "react";
 import Loading from "@/app/(keep)/loading";
-import PageClient from "../builder/page.client";
+import { WorkflowBuilderPageClient } from "../builder/page.client";
 import Link from "next/link";
 
 type PageProps = {
@@ -28,8 +28,8 @@ export default function Page({ params, searchParams }: PageProps) {
     <>
       {!workflowPreviewData && <Loading />}
       {workflowPreviewData && workflowPreviewData.name === key && (
-        <PageClient
-          workflow={workflowPreviewData?.Workflow_raw}
+        <WorkflowBuilderPageClient
+          workflowRaw={workflowPreviewData?.Workflow_raw}
           workflowId={params?.workflowId}
           isPreview={true}
         />