diff --git a/tools/data_source/upload.py b/tools/data_source/upload.py
index 0802f6a3dd0..838aa069a8f 100644
--- a/tools/data_source/upload.py
+++ b/tools/data_source/upload.py
@@ -12,6 +12,7 @@ import shutil
 import sys
 import tempfile
 import zipfile
+import cProfile
 from json import dumps, loads
 
 from six.moves.urllib.request import urlopen