diff --git a/flo_ai/flo_ai/arium/builder.py b/flo_ai/flo_ai/arium/builder.py
index cc66bdb8..6071f7b6 100644
--- a/flo_ai/flo_ai/arium/builder.py
+++ b/flo_ai/flo_ai/arium/builder.py
@@ -457,6 +457,9 @@ def from_yaml(
                         f'Either provide the agent in the agents parameter or add configuration fields.'
                     )
 
+            elif agents and agent_name in agents:
+                agent = agents[agent_name]
+
             # Method 2: Direct agent definition
             elif (
                 'job' in agent_config