diff --git a/skyvern-frontend/src/routes/workflows/workflowRun/BlockCard.tsx b/skyvern-frontend/src/routes/workflows/workflowRun/BlockCard.tsx
index fc24664a..e98da892 100644
--- a/skyvern-frontend/src/routes/workflows/workflowRun/BlockCard.tsx
+++ b/skyvern-frontend/src/routes/workflows/workflowRun/BlockCard.tsx
@@ -1,10 +1,8 @@
-import { ExternalLinkIcon } from "@radix-ui/react-icons";
+import { CubeIcon } from "@radix-ui/react-icons";
 import { WorkflowRunBlock } from "../types/workflowRunTypes";
 import { cn } from "@/util/utils";
 import { WorkflowBlockIcon } from "../editor/nodes/WorkflowBlockIcon";
 import { workflowBlockTitle } from "../editor/nodes/types";
-import { Button } from "@/components/ui/button";
-import { Link } from "react-router-dom";
 
 type Props = {
   active: boolean;
@@ -23,7 +21,7 @@ function BlockCard({ block, onClick, active }: Props) {
       )}
       onClick={onClick}
     >
-      <div className="flex items-center justify-between">
+      <div className="flex justify-between">
         <div className="flex gap-3">
           <WorkflowBlockIcon
             workflowBlockType={block.block_type}
@@ -31,18 +29,10 @@ function BlockCard({ block, onClick, active }: Props) {
           />
           <span>{workflowBlockTitle[block.block_type]}</span>
         </div>
-        {block.task_id && (
-          <Button
-            title="Go to task diagnostics"
-            asChild
-            size="icon"
-            className="size-8 bg-slate-800 text-primary hover:bg-slate-700"
-          >
-            <Link to={`/tasks/${block.task_id}/diagnostics`}>
-              <ExternalLinkIcon />
-            </Link>
-          </Button>
-        )}
+        <div className="flex items-center gap-1 rounded bg-slate-elevation5 px-2 py-1">
+          <CubeIcon className="size-4" />
+          <span className="text-xs">Block</span>
+        </div>
       </div>
     </div>
   );
diff --git a/skyvern-frontend/src/routes/workflows/workflowRun/WorkflowRunTimelineItemInfoSection.tsx b/skyvern-frontend/src/routes/workflows/workflowRun/WorkflowRunTimelineItemInfoSection.tsx
index 73f88401..86ef35b8 100644
--- a/skyvern-frontend/src/routes/workflows/workflowRun/WorkflowRunTimelineItemInfoSection.tsx
+++ b/skyvern-frontend/src/routes/workflows/workflowRun/WorkflowRunTimelineItemInfoSection.tsx
@@ -13,6 +13,8 @@ import { WorkflowBlockTypes } from "../types/workflowTypes";
 import { statusIsAFailureType } from "@/routes/tasks/types";
 import { SendEmailBlockInfo } from "./blockInfo/SendEmailBlockInfo";
 import { WorkflowRunOverviewActiveElement } from "./WorkflowRunOverview";
+import { ExternalLinkIcon } from "@radix-ui/react-icons";
+import { Link } from "react-router-dom";
 
 type Props = {
   activeItem: WorkflowRunOverviewActiveElement;
@@ -84,6 +86,17 @@ function WorkflowRunTimelineItemInfoSection({ activeItem }: Props) {
               )}
               <TabsTrigger value="navigation_goal">Navigation Goal</TabsTrigger>
               <TabsTrigger value="parameters">Parameters</TabsTrigger>
+              {item.task_id && (
+                <Link
+                  to={`/tasks/${item.task_id}/diagnostics`}
+                  title="Go to diagnostics"
+                >
+                  <div className="flex items-center gap-2 px-3 py-1 text-sm font-medium">
+                    <ExternalLinkIcon />
+                    <span>Diagnostics</span>
+                  </div>
+                </Link>
+              )}
             </TabsList>
             {item.status === Status.Completed && (
               <TabsContent value="extracted_information">