diff --git a/server/assignments/schema/queries.py b/server/assignments/schema/queries.py
index 56e1fe0e..a62b7fb9 100644
--- a/server/assignments/schema/queries.py
+++ b/server/assignments/schema/queries.py
@@ -20,5 +20,5 @@ class StudentSubmissionQuery:
         id = kwargs.get('id')
         submission = get_object(StudentSubmission, id)
         if user == submission.student or (user.is_teacher() and submission.student in user.users_in_active_school_class()):
-            return root
+            return submission
         return None