diff --git a/ELMo/train.py b/ELMo/train.py
index c062df304c08e52736f3526e1da8ee06456c8773..1bd8f36eb3c63d619d3f093a35754c48d8cc5a0d 100755
--- a/ELMo/train.py
+++ b/ELMo/train.py
@@ -18,6 +18,7 @@ from __future__ import print_function
 
 import six
 import numpy as np
+import random
 import time
 import os
 import math