diff --git a/assignments/A2/a2_game_tree.py b/assignments/A2/a2_game_tree.py
index a4b4d73..8e77246 100644
--- a/assignments/A2/a2_game_tree.py
+++ b/assignments/A2/a2_game_tree.py
@@ -89,7 +89,7 @@ class GameTree:
         return None
 
     def add_subtree(self, subtree: GameTree) -> None:
-        """Add a subtree to this game tree."""
+        """Add a subtree to this game tree and updates white win probability."""
         self._subtrees.append(subtree)
         self._update_white_win_probability()
 
diff --git a/assignments/A2/a2_part1.py b/assignments/A2/a2_part1.py
index 69023d6..6b8f590 100644
--- a/assignments/A2/a2_part1.py
+++ b/assignments/A2/a2_part1.py
@@ -167,7 +167,7 @@ def part1_runner(games_file: str, n: int, black_random: bool) -> None:
     """
     tree = load_game_tree(games_file)
     white = RandomTreePlayer(tree)
-    black = a2_minichess.RandomPlayer() if black_random else white
+    black = a2_minichess.RandomPlayer() if black_random else RandomTreePlayer(tree)
     a2_minichess.run_games(n, white, black, show_stats=True)