diff --git a/zesje/pregrader.py b/zesje/pregrader.py
index 5fd216f6f7d7a9f199769ef6bc806ff5c3922421..2339d325ce78b345a66bd1678fa7d3b3d4d37937 100644
--- a/zesje/pregrader.py
+++ b/zesje/pregrader.py
@@ -29,6 +29,7 @@ def pregrade(exam_token, image):
 
 def add_feedback_to_solution(page_img, barcode):
     exam = Exam.query.filter(Exam.token == barcode.token).first()
+    sub = Submission.query.filter(Submission.copy_number == barcode.copy, Submission.exam_id == exam.id).one_or_none()
 
     problems = exam.problems
     problems_on_page = list(filter(lambda p: p.widget.page == barcode.page, problems))
@@ -40,7 +41,7 @@ def add_feedback_to_solution(page_img, barcode):
             # check width and so forth
 
             if box_is_filled(box, page_img):
-                problem.solution.feedback = mc_option.feedback
+                sub.feedback = mc_option.feedback
                 db.session.commit()