diff --git a/grader b/grader
--- a/grader
+++ b/grader
@@ -129,10 +129,12 @@
     @users = {}
   end
 
-  def after_save_hook(user, problem, grading_result)
+  def after_save_hook(submission, grading_result)
   end
 
-  def save(user, problem, grading_result)
+  def save(submission, grading_result)
+    user = submission.user
+    problem = submission.problem
     if not @problems.has_key? problem.id
       @problems[problem.id] = problem
     end
@@ -141,7 +143,7 @@
     end
     @results[[user.id, problem.id]] = grading_result
 
-    after_save_hook(user, problem, grading_result)
+    after_save_hook(submission, grading_result)
   end
 
   def print_report_by_user
diff --git a/lib/submission_helper.rb b/lib/submission_helper.rb
--- a/lib/submission_helper.rb
+++ b/lib/submission_helper.rb
@@ -42,8 +42,7 @@
     def report(sub,test_result_dir)
       result = read_result(test_result_dir)
       if @result_collector
-        @result_collector.save(sub.user, 
-                               sub.problem,
+        @result_collector.save(sub,                               
                                result)
       end
       save_result(sub,result)