diff --git a/Season2/1 Word2Vec/word2vec_tf.py b/Season2/1 Word2Vec/word2vec_tf.py
index 0049ecb..f00e7a3 100644
--- a/Season2/1 Word2Vec/word2vec_tf.py	
+++ b/Season2/1 Word2Vec/word2vec_tf.py	
@@ -183,7 +183,12 @@ def generate_batch(batch_size, num_skips, skip_window):
     similarity = tf.matmul(valid_embeddings, normalized_embeddings, transpose_b=True)
 
     # Add variable initializer.
-    init = tf.global_variables_initializer()
+    tf_major_ver = int(tf.__version__.split(".")[0])
+    tf_minor_ver = int(tf.__version__.split(".")[1])
+    if(tf_major_ver==0 and tf_minor_ver<12):
+      init = tf.initialize_all_variables()
+    else:
+      init = tf.global_variables_initializer()
 
   # Step 5: Begin training.
   num_steps = 100001