diff --git a/src/Nncase.Schedule/Schedule/MonteCarloTreeSearch/SearchNode.cs b/src/Nncase.Schedule/Schedule/MonteCarloTreeSearch/SearchNode.cs
index 6770704ef..f8fa081e3 100644
--- a/src/Nncase.Schedule/Schedule/MonteCarloTreeSearch/SearchNode.cs
+++ b/src/Nncase.Schedule/Schedule/MonteCarloTreeSearch/SearchNode.cs
@@ -41,4 +41,6 @@ public SearchNode(SearchNode<T> parent, IEnvironmentState<T> state)
     public bool IsRootNode => Parent is null;
 
     public abstract void Update(double reward);
+
+    public abstract void Dump(System.CodeDom.Compiler.IndentedTextWriter writer);
 }
diff --git a/src/Nncase.Schedule/Schedule/TileGraph/GraphMCTS.cs b/src/Nncase.Schedule/Schedule/TileGraph/GraphMCTS.cs
index 57d74bd66..9580a9f8a 100644
--- a/src/Nncase.Schedule/Schedule/TileGraph/GraphMCTS.cs
+++ b/src/Nncase.Schedule/Schedule/TileGraph/GraphMCTS.cs
@@ -153,7 +153,7 @@ public void Dump(string name)
         }
     }
 
-    public void Dump(System.CodeDom.Compiler.IndentedTextWriter writer)
+    public override void Dump(System.CodeDom.Compiler.IndentedTextWriter writer)
     {
         writer.WriteLine($"- name: {this}");
         writer.WriteLine($"  Action: {Action}");
@@ -161,7 +161,7 @@ public void Dump(System.CodeDom.Compiler.IndentedTextWriter writer)
         writer.WriteLine($"  VisitTimes: {VisitTimes}");
         writer.WriteLine($"  Children:");
         writer.Indent += 1;
-        foreach (var item in Children.OfType<MCTNode>())
+        foreach (var item in Children)
         {
             item.Dump(writer);
         }