diff --git a/dowhy/graph.py b/dowhy/graph.py
index 958a829ea1..7f505ea683 100644
--- a/dowhy/graph.py
+++ b/dowhy/graph.py
@@ -5,11 +5,10 @@
 """
 
 from abc import abstractmethod
-from typing import Any, List
+from typing import Any, List, Protocol
 
 import networkx as nx
 from networkx.algorithms.dag import has_cycle
-from typing_extensions import Protocol
 
 
 class HasNodes(Protocol):