diff --git a/server/news/tests/test_newsteaser.py b/server/news/tests/test_newsteaser.py
index deff79ae..9a7ec8a9 100644
--- a/server/news/tests/test_newsteaser.py
+++ b/server/news/tests/test_newsteaser.py
@@ -1,14 +1,15 @@
-from unittest import TestCase
+from django.test import TestCase
 from graphene.test import Client
 
 from api.schema import schema
 from news.factories import NewsTeaserFactory
+from news.models import NewsTeaser
 
 
 class NewsTeaserTests(TestCase):
     def setUp(self):
-        self.news_teaser1 = NewsTeaserFactory(order_id=1)
-        self.news_teaser2 = NewsTeaserFactory(order_id=2)
+        self.news_teaser1 = NewsTeaserFactory(order_id=1, title="should come first")
+        self.news_teaser2 = NewsTeaserFactory(order_id=2, title="should be second")
 
         self.client = Client(schema=schema)
 
@@ -16,17 +17,13 @@ class NewsTeaserTests(TestCase):
         query = '''
             query NewsTeasers {
                 newsTeasers {
-                  edges {
-                    node {
-                      id
-                      description
-                      title
-                      imageUrl
-                      newsArticleUrl
-                      displayDate
-                      imageSource
-                    }
-                  }
+                  id
+                  description
+                  title
+                  imageUrl
+                  newsArticleUrl
+                  displayDate
+                  imageSource
                 }
             }
         '''
@@ -36,7 +33,7 @@ class NewsTeaserTests(TestCase):
     def test_can_get_news_teaser(self):
         result = self.make_query()
         self.assertIsNone(result.get('errors'))
-        news_teasers = result.get('data').get('newsTeasers').get('edges')
-        self.assertEqual(news_teasers[0].get('node').get('title'), self.news_teaser1.title)
-        self.assertEqual(news_teasers[1].get('node').get('title'), self.news_teaser2.title)
+        news_teasers = result.get('data').get('newsTeasers')
+        self.assertEqual(news_teasers[0].get('title'), self.news_teaser1.title)
+        self.assertEqual(news_teasers[1].get('title'), self.news_teaser2.title)