From 79dd717753a1df074d9324ae9af89ccd4ade07db Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Marcin=20G=C4=99bala?= <maarcin.gebala@gmail.com>
Date: Wed, 20 Nov 2019 10:31:33 +0100
Subject: [PATCH] Fix tests specific to demo

---
 tests/api/test_graphql.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/tests/api/test_graphql.py b/tests/api/test_graphql.py
index 5033d583002..ee09a026a44 100644
--- a/tests/api/test_graphql.py
+++ b/tests/api/test_graphql.py
@@ -17,7 +17,9 @@
 from tests.api.utils import get_graphql_content
 
 
-def test_middleware_dont_generate_sql_requests(client, settings, assert_num_queries):
+def test_middleware_dont_generate_sql_requests(
+    client, settings, django_assert_num_queries
+):
     """When requesting on the GraphQL API endpoint, no SQL request should happen
     indirectly. This test ensures that."""