diff --git a/server/books/tests/test_note_mutations.py b/server/books/tests/test_note_mutations.py
index 5b5014eb..4ad29fdd 100644
--- a/server/books/tests/test_note_mutations.py
+++ b/server/books/tests/test_note_mutations.py
@@ -13,7 +13,7 @@ from notes.models import Note, ModuleBookmark, ChapterBookmark
 
 class NoteMutationTestCase(TestCase):
     def setUp(self):
-        self.module = ModuleFactory()
+        self.module = ModuleFactory(slug='some-module-slug')
         self.chapter = Chapter(title='Hello')
         self.module.add_child(instance=self.chapter)
         self.user = user = UserFactory(username='aschi')
@@ -53,8 +53,9 @@ class NewNoteMutationTestCase(NoteMutationTestCase):
         result = self.client.execute(self.add_mutation, variables={
             'input': {
                 'note': {
-                    'parent': to_global_id('ModuleNode', self.module.pk),
-                    'text': text
+                    'parent': self.module.slug,
+                    'text': text,
+                    'type': 'module'
                 }
             }
         })