diff --git a/libs/designer-v2/src/lib/core/state/designerView/designerViewSelectors.ts b/libs/designer-v2/src/lib/core/state/designerView/designerViewSelectors.ts
index 07e162cec0d..4b12dafa60a 100644
--- a/libs/designer-v2/src/lib/core/state/designerView/designerViewSelectors.ts
+++ b/libs/designer-v2/src/lib/core/state/designerView/designerViewSelectors.ts
@@ -1,4 +1,4 @@
-import { equals, SUBGRAPH_TYPES } from '@microsoft/logic-apps-shared';
+import { equals } from '@microsoft/logic-apps-shared';
 import type { RootState } from '../../store';
 import { useSelector } from 'react-redux';
 import { isA2AWorkflow } from '../workflow/helper';
@@ -40,6 +40,6 @@ export const useIsA2AWorkflow = () => {
 
 export const useWorkflowHasAgentLoop = () => {
   return useSelector((state: RootState) => {
-    return Object.values(state.workflow.nodesMetadata).some((node) => node.subgraphType === SUBGRAPH_TYPES.AGENT_CONDITION);
+    return Object.values(state.workflow.operations).some((operation) => equals(operation.type, 'agent'));
   });
 };