diff --git a/consumer_api/tests/consumers_view_test.py b/consumer_api/tests/consumers_view_test.py
index fc1a9be..7d8c8ea 100644
--- a/consumer_api/tests/consumers_view_test.py
+++ b/consumer_api/tests/consumers_view_test.py
@@ -8,7 +8,7 @@ def client():
     current_app.config['TESTING'] = True
 
     with current_app.test_client() as client:
-        yield
+        yield client
 
 def test_response_length(client):
     r = client.post('/log')
diff --git a/consumer_api/tests/log_view_test.py b/consumer_api/tests/log_view_test.py
index f102664..2b619b9 100644
--- a/consumer_api/tests/log_view_test.py
+++ b/consumer_api/tests/log_view_test.py
@@ -8,7 +8,7 @@ def client():
     current_app.config['TESTING'] = True
 
     with current_app.test_client() as client:
-        yield
+        yield client
 
 def test_log_code_get(client):
     r = client.get('/log')