diff --git a/python/packages/core/agent_framework/_workflows/_edge.py b/python/packages/core/agent_framework/_workflows/_edge.py
index 549bd9c7476..a9483f7b853 100644
--- a/python/packages/core/agent_framework/_workflows/_edge.py
+++ b/python/packages/core/agent_framework/_workflows/_edge.py
@@ -5,6 +5,7 @@
 import inspect
 import logging
 import uuid
+from builtins import type as builtin_type
 from collections.abc import Awaitable, Callable, Sequence
 from dataclasses import dataclass, field
 from typing import Any, ClassVar, TypeAlias, TypeVar
@@ -312,8 +313,6 @@ class EdgeGroup(DictConvertible):
     type: str
     edges: list[Edge]
 
-    from builtins import type as builtin_type
-
     _TYPE_REGISTRY: ClassVar[dict[str, builtin_type[EdgeGroup]]] = {}
 
     def __init__(