diff --git a/client/cypress/integration/frontend/assignment.spec.js b/client/cypress/integration/frontend/assignment.spec.js
index 9dcbbd7c..00926c40 100644
--- a/client/cypress/integration/frontend/assignment.spec.js
+++ b/client/cypress/integration/frontend/assignment.spec.js
@@ -26,12 +26,15 @@ describe('Assignments', () => {
     },
     ModuleDetailsQuery: {},
     StudentSubmissions: {
-      studentSubmission
+      studentSubmission,
     },
-    UpdateSubmissionFeedback: {
-      updateSubmissionFeedback: {
-        successful: true,
-      },
+    UpdateSubmissionFeedback({input: {submissionFeedback}}) {
+      return {
+        updateSubmissionFeedback: {
+          successful: true,
+          updatedSubmissionFeedback: submissionFeedback,
+        },
+      };
     },
   };