diff --git a/core/src/test/java/org/elasticsearch/index/query/TemplateQueryBuilderTests.java b/core/src/test/java/org/elasticsearch/index/query/TemplateQueryBuilderTests.java
index e18a0f83d21..8dc2a61c110 100644
--- a/core/src/test/java/org/elasticsearch/index/query/TemplateQueryBuilderTests.java
+++ b/core/src/test/java/org/elasticsearch/index/query/TemplateQueryBuilderTests.java
@@ -57,7 +57,7 @@ public class TemplateQueryBuilderTests extends BaseQueryTestCase<TemplateQueryBu
 
     @Override
     protected void doAssertLuceneQuery(TemplateQueryBuilder queryBuilder, Query query, QueryShardContext context) throws IOException {
-        assertEquals(templateBase.toQuery(createShardContext()), query);
+        assertEquals(templateBase.toQuery(context), query);
     }
 
     @Test