diff --git a/hw1/tf_util.py b/hw1/tf_util.py
index c1427957b..9aaf6b12f 100644
--- a/hw1/tf_util.py
+++ b/hw1/tf_util.py
@@ -88,8 +88,8 @@ def make_session(num_cpu):
 
 ALREADY_INITIALIZED = set()
 def initialize():
-    new_variables = set(tf.all_variables()) - ALREADY_INITIALIZED
-    get_session().run(tf.initialize_variables(new_variables))
+    new_variables = set(tf.global_variables()) - ALREADY_INITIALIZED
+    get_session().run(tf.variables_initializer(new_variables))
     ALREADY_INITIALIZED.update(new_variables)