diff --git a/src/bots/mcts_bot.rs b/src/bots/mcts_bot.rs
index 8d0592a..ad2c8e7 100644
--- a/src/bots/mcts_bot.rs
+++ b/src/bots/mcts_bot.rs
@@ -114,8 +114,8 @@ impl<'a> ArenaTree<'a> {
             .children
             .iter()
             .max_by(|l, r| {
-                let lv = self.arena[*l.to_owned()].data.visited;
-                let rv = self.arena[*r.to_owned()].data.visited;
+                let lv = self.arena[*l.to_owned()].data.average_reward();
+                let rv = self.arena[*r.to_owned()].data.average_reward();
                 lv.partial_cmp(&rv).unwrap()
             })
             .unwrap();