diff --git a/expo/scripts/visualize_experiment.py b/expo/scripts/visualize_experiment.py
index 42b4490ec..e2443d0fd 100644
--- a/expo/scripts/visualize_experiment.py
+++ b/expo/scripts/visualize_experiment.py
@@ -20,4 +20,4 @@ if __name__ == "__main__":
     root = mcts.root_node
     G = nx.DiGraph()
     build_tree_recursive(G, "0", root)
-    visualize_tree(G, save_path="results/tree.png")
+    visualize_tree(G, save_path=f"results/{args.task}-tree.png")