diff --git a/app/views/main/_problem.html.haml b/app/views/main/_problem.html.haml
--- a/app/views/main/_problem.html.haml
+++ b/app/views/main/_problem.html.haml
@@ -10,7 +10,7 @@
     = link_to "[subs]", main_submission_path(problem.id)
   %td
     = render :partial => 'submission_short',
-      :locals => {:submission => @prob_submissions[problem.id][:submission], :problem_name => problem.name }
+      :locals => {:submission => @prob_submissions[problem.id][:submission], :problem_name => problem.name, :problem_id => problem.id }
   %td
     - if @prob_submissions[problem.id][:submission]
       = link_to 'Edit', direct_edit_submission_path(@prob_submissions[problem.id][:submission]), class: 'btn btn-success'