diff --git a/pic-sure-api-war/src/test/java/edu/harvard/dbmi/avillach/PicsureQueryServiceTest.java b/pic-sure-api-war/src/test/java/edu/harvard/dbmi/avillach/PicsureQueryServiceTest.java
index b9d245bd..90ed6cec 100644
--- a/pic-sure-api-war/src/test/java/edu/harvard/dbmi/avillach/PicsureQueryServiceTest.java
+++ b/pic-sure-api-war/src/test/java/edu/harvard/dbmi/avillach/PicsureQueryServiceTest.java
@@ -385,7 +385,7 @@ public void testQuerySyncValidNoResponseId() {
 
 		// Return mocks when needed
 		Response resp = mock(Response.class);
-		when(webClient.querySync(any(), any())).thenReturn(resp);
+		when(webClient.querySync(any(), any(), any())).thenReturn(resp);
 
 		// Mock persisting the queryentity, so that it has an ID and we can test that
 		// the correct information is stored in it
@@ -439,7 +439,7 @@ public void testQuerySyncValidWithResponseId() {
 		MultivaluedMap<String, Object> headerMap= new MultivaluedHashMap<String, Object>();
 		headerMap.add(ResourceWebClient.QUERY_METADATA_FIELD, resultId);
 		when(resp.getHeaders()).thenReturn(headerMap);
-		when(webClient.querySync(any(), any())).thenReturn(resp);
+		when(webClient.querySync(any(), any(), any())).thenReturn(resp);
 
 		// Mock persisting the queryentity, so that it has an ID and we can test that
 		// the correct information is stored in it