diff --git a/src/components/shared/TaskDetails/Actions.tsx b/src/components/shared/TaskDetails/Actions.tsx
index 481c665cd..b273d4e57 100644
--- a/src/components/shared/TaskDetails/Actions.tsx
+++ b/src/components/shared/TaskDetails/Actions.tsx
@@ -1,6 +1,7 @@
 import { useState } from "react";
 import { FaPython } from "react-icons/fa";
 
+import { BlockStack } from "@/components/ui/layout";
 import useToastNotification from "@/hooks/useToastNotification";
 import type { ComponentSpec } from "@/utils/componentSpec";
 import {
@@ -110,11 +111,14 @@ const TaskActions = ({
     },
   ];
 
-  const allActions: Action[] = [...customActions, ...sharedActions];
-
   return (
     <>
-      <ActionBlock actions={allActions} className={className} />
+      <BlockStack gap="4" className={className}>
+        <ActionBlock title="Component Actions" actions={sharedActions} />
+        {customActions.length > 0 && (
+          <ActionBlock title="Node Actions" actions={customActions} />
+        )}
+      </BlockStack>
 
       {isEditDialogOpen && (
         <ComponentEditorDialog