diff --git a/run.py b/run.py
index 4e39267..66f4c56 100644
--- a/run.py
+++ b/run.py
@@ -8,6 +8,7 @@
 from utils import *
 import time
 from pickle import NONE
+import datetime
 
 start_time = time.time()