diff --git a/models/onnx_export.py b/models/onnx_export.py
index d17f8b2..fe0287e 100644
--- a/models/onnx_export.py
+++ b/models/onnx_export.py
@@ -9,6 +9,7 @@ import argparse
 import onnx
 
 from models.common import *
+from utils import google_utils
 
 if __name__ == '__main__':
     parser = argparse.ArgumentParser()