diff --git a/train.py b/train.py
index 0c4e1d9..c264514 100644
--- a/train.py
+++ b/train.py
@@ -101,7 +101,7 @@ def train_save_evaluate(params, kb, train_set, dev_set, ind2emoji, embeddings_ar
 
         else:
             # For visualizing using tensorboard
-            summary_writer = tf.train.SummaryWriter(model_folder + '/board', graph=sess.graph)
+            summary_writer = tf.summary.FileWriter(model_folder + '/board', graph=sess.graph)
 
             # Keep track of how the model is training
             hooks = [