From 883848a15cd0ca5f51445fb04b8e9a1d12dff1e5 Mon Sep 17 00:00:00 2001
From: Daniel Gallagher <daniellg@yelp.com>
Date: Mon, 28 May 2018 14:41:02 -0700
Subject: [PATCH] Fix test to pass once more

---
 graphene/tests/issues/test_356.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/graphene/tests/issues/test_356.py b/graphene/tests/issues/test_356.py
index 3755a8ef..9d8fc77b 100644
--- a/graphene/tests/issues/test_356.py
+++ b/graphene/tests/issues/test_356.py
@@ -21,9 +21,11 @@ class MyUnion(graphene.Union):
 
 
 def test_issue():
+    class Query(graphene.ObjectType):
+        things = relay.ConnectionField(MyUnion)
+
     with pytest.raises(Exception) as exc_info:
-        class Query(graphene.ObjectType):
-            things = relay.ConnectionField(MyUnion)
+        graphene.Schema(query=Query)
 
     assert str(exc_info.value) == (
         'IterableConnectionField type have to be a subclass of Connection. '