diff --git a/frontend/src/components/submission_notes/SubmissionCorrection.vue b/frontend/src/components/submission_notes/SubmissionCorrection.vue
index c520dc35dae038cb45824e358d0c50326e9cc624..4687c239cc04d768c0491858aaa6c2ccd5e95293 100644
--- a/frontend/src/components/submission_notes/SubmissionCorrection.vue
+++ b/frontend/src/components/submission_notes/SubmissionCorrection.vue
@@ -172,8 +172,12 @@ export default {
       }, 5e3)
     },
     hasHiddenComment (lineNo) {
-      return !this.showFeedback && this.origFeedback[lineNo]
+      const hasOrigComment = this.origFeedback && this.origFeedback[lineNo]
         && this.origFeedback[lineNo].length > 0
+
+      const hasUpdatedComment = this.updatedFeedback && this.updatedFeedback[lineNo]
+
+      return !this.showFeedback && (hasOrigComment || hasUpdatedComment)
     },
     init () {
       SubmissionNotes.RESET_STATE()