diff --git a/dowhy/graph_learners/cdt.py b/dowhy/graph_learners/cdt.py
index 8c96a42f7f..fb654bd020 100644
--- a/dowhy/graph_learners/cdt.py
+++ b/dowhy/graph_learners/cdt.py
@@ -1,5 +1,8 @@
+import networkx as nx
+import numpy as np
+
 from dowhy.graph_learners import GraphLearner
-from dowhy.utils.graph_operations import *
+from dowhy.utils.graph_operations import adjacency_matrix_to_graph, str_to_dot
 
 from . import get_library_class_object
 
diff --git a/dowhy/graph_learners/ges.py b/dowhy/graph_learners/ges.py
index 043b91ccfe..1f31d35d28 100644
--- a/dowhy/graph_learners/ges.py
+++ b/dowhy/graph_learners/ges.py
@@ -1,7 +1,9 @@
 from importlib import import_module
 
+import numpy as np
+
 from dowhy.graph_learners import GraphLearner
-from dowhy.utils.graph_operations import *
+from dowhy.utils.graph_operations import adjacency_matrix_to_graph, str_to_dot
 
 
 class GES(GraphLearner):
diff --git a/dowhy/graph_learners/lingam.py b/dowhy/graph_learners/lingam.py
index a6c4cb8713..47860164dc 100644
--- a/dowhy/graph_learners/lingam.py
+++ b/dowhy/graph_learners/lingam.py
@@ -1,5 +1,7 @@
+import numpy as np
+
 from dowhy.graph_learners import GraphLearner
-from dowhy.utils.graph_operations import *
+from dowhy.utils.graph_operations import adjacency_matrix_to_graph, str_to_dot
 
 from . import get_library_class_object