From f4729b70396dc5e7ab1aa3235781910c9ca0ed6d Mon Sep 17 00:00:00 2001
From: narawat lamaiin <narawat@outlook.com>
Date: Mon, 6 May 2024 20:46:34 +0700
Subject: [PATCH] update

---
 src/mcts.jl | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/mcts.jl b/src/mcts.jl
index 95e698e..c4d8d47 100644
--- a/src/mcts.jl
+++ b/src/mcts.jl
@@ -452,13 +452,14 @@ function runMCTS(
     if node.isterminal
       # MCTS arrive at the leaf node that is also a terminal state,
       # do nothing then go directly to backpropagation
+      backpropagate(leafNode, node.reward)
     else
       expand(a, node, decisionMaker, progressValueEstimator, isterminal, n=n)
       leafNode = UCTselect(node, w)
       simTrajectoryReward = simulate(a, leafNode, decisionMaker, progressValueEstimator,
                                     isterminal, maxDepth, n=n)
+      backpropagate(leafNode, simTrajectoryReward)
     end
-    backpropagate(leafNode, simTrajectoryReward)
   end
 
   best_child_state = argmax([child.total_reward / child.visits for child in values(root.children)])