diff --git a/tests/test_discovery.py b/tests/test_discovery.py
index 92213ac..b8b5ac0 100644
--- a/tests/test_discovery.py
+++ b/tests/test_discovery.py
@@ -41,7 +41,6 @@
   from cgi import parse_qs
 
 
-from apiclient.discovery import _add_query_parameter
 from apiclient.discovery import _fix_up_media_upload
 from apiclient.discovery import _fix_up_method_description
 from apiclient.discovery import _fix_up_parameters
@@ -68,6 +67,7 @@
 from oauth2client import GOOGLE_TOKEN_URI
 from oauth2client.anyjson import simplejson
 from oauth2client.client import OAuth2Credentials
+from oauth2client.util import _add_query_parameter
 import uritemplate