diff --git a/apps/sim/app/workspace/[workspaceId]/w/[workflowId]/workflow.tsx b/apps/sim/app/workspace/[workspaceId]/w/[workflowId]/workflow.tsx
index 07bb0c143..78d132770 100644
--- a/apps/sim/app/workspace/[workspaceId]/w/[workflowId]/workflow.tsx
+++ b/apps/sim/app/workspace/[workspaceId]/w/[workflowId]/workflow.tsx
@@ -57,7 +57,6 @@ import {
   estimateBlockDimensions,
   filterProtectedBlocks,
   getClampedPositionForNode,
-  hasProtectedBlocks,
   isBlockProtected,
   isEdgeProtected,
   isInEditableElement,
@@ -3606,10 +3605,7 @@ const WorkflowContent = React.memo(() => {
               canRunFromBlock={runFromBlockState.canRun}
               disableEdit={
                 !effectivePermissions.canEdit ||
-                hasProtectedBlocks(
-                  contextMenuBlocks.map((b) => b.id),
-                  blocks
-                )
+                contextMenuBlocks.some((b) => b.locked || b.isParentLocked)
               }
               userCanEdit={effectivePermissions.canEdit}
               isExecuting={isExecuting}