diff --git a/client/src/components/content-blocks/assignment/Assignment.vue b/client/src/components/content-blocks/assignment/Assignment.vue
index e420ed44..13fe2c3f 100644
--- a/client/src/components/content-blocks/assignment/Assignment.vue
+++ b/client/src/components/content-blocks/assignment/Assignment.vue
@@ -115,6 +115,18 @@ graphql(`
   }
 `);
 
+graphql(`
+  fragment AssignmentParts on AssignmentNode {
+    id
+    title
+    assignment
+    solution
+    submission {
+      ...SubmissionParts
+    }
+  }
+`);
+
 const { me } = getMe();
 
 const { result, onResult } = useQuery(
@@ -362,9 +374,9 @@ const saveInput = (answer: string) => {
   corrections.value = '';
   unsaved.value = true;
   /*
-              We update the assignment on this component, so the changes are reflected on it. The server does not return
-              the updated entity, to prevent the UI to update when the user is entering his input
-             */
+            We update the assignment on this component, so the changes are reflected on it. The server does not return
+            the updated entity, to prevent the UI to update when the user is entering his input
+           */
   assignment.value.submission.text = answer;
   _save(assignment.value.submission);