diff --git a/monte_carlo_tree_search.py b/monte_carlo_tree_search.py
index 17c8377..8677a84 100644
--- a/monte_carlo_tree_search.py
+++ b/monte_carlo_tree_search.py
@@ -1,4 +1,4 @@
-import torch
+
 import math
 import numpy as np
 
@@ -137,7 +137,7 @@ def run(self, model, state, to_play):
 
         return root
 
-    def backpropagate(self, search_path, value, to_play):
+    def backup(self, search_path, value, to_play):
         """
         At the end of a simulation, we propagate the evaluation all the way up the tree
         to the root.