diff --git a/utils/create_data.py b/utils/create_data.py
index 07e23ce..680b3af 100644
--- a/utils/create_data.py
+++ b/utils/create_data.py
@@ -25,9 +25,9 @@
 from io import BytesIO
 import os
 import pickle
-import StringIO
+from io import StringIO
 import tarfile
-import urllib2
+from urllib.request import urlopen
 
 import keras.backend as K
 from keras.datasets import cifar10