diff --git a/tools/check_for_lite.py b/tools/check_for_lite.py
index 94581b20ccdd43fd4f5b39c7210b3a74b2e273d1..78fb55f5ee309106daf12d805a2efa95dcdaaad2 100644
--- a/tools/check_for_lite.py
+++ b/tools/check_for_lite.py
@@ -1,4 +1,4 @@
-import urllib
+from six.moves import urllib
 import sys
 from paddle.fluid.framework import Program