diff --git a/create_tfrecords.py b/create_tfrecords.py
index 608bfd3..3ba1137 100644
--- a/create_tfrecords.py
+++ b/create_tfrecords.py
@@ -25,6 +25,7 @@
 
 import argparse
 from datetime import datetime
+import json
 import os
 from Queue import Queue
 import random