diff --git a/lib/core/update.py b/lib/core/update.py
index dff882a..408a2b1 100755
--- a/lib/core/update.py
+++ b/lib/core/update.py
@@ -7,7 +7,10 @@
 
 import os
 import sys
-import urllib2
+try:
+    import urllib2
+except ImportError:
+    import urllib.request as urllib2
 import tarfile
 
 from lib.common.utils import checksum