diff --git a/app/controllers/student_work_controller.rb b/app/controllers/student_work_controller.rb
index 3e1e0ba29..01f858442 100644
--- a/app/controllers/student_work_controller.rb
+++ b/app/controllers/student_work_controller.rb
@@ -387,8 +387,8 @@ class StudentWorkController < ApplicationController
     if stundet_work && params[:results] && params[:results].class.to_s == "Array"
       homework_common = stundet_work.homework_common
       params[:results].each do |result|
-        homework_test = homework_common.homework_tests.where("input = '#{result[:input]}' AND output = '#{result[:output]}'").first
-        if homework_test
+        homework_tests = homework_common.homework_tests.where("input = '#{result[:input]}' AND output = '#{result[:output]}'")
+        homework_tests.each do |homework_test|
           student_work_test = StudentWorkTest.new
           student_work_test.student_work = stundet_work
           student_work_test.homework_test = homework_test