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
@@ -2,7 +2,7 @@
   .problem-form{:id => "problem-form-#{problem.id}"}
     - form_tag({ :action => 'download_input', :id => problem.id }, :method => :post) do
       %b Input:
-      %input{:type => "submit", :value => "Download input"}
+      %input{:type => "submit", :value => "Download input", :onclick => "CodejomTimeout.updateTimeoutAfterDownloadClick(#{problem.id}); return true;"}
       %span{:id => "problem-timing-message-#{problem.id}"}
         = "After downloading, you have #{TEST_ASSIGNMENT_EXPIRATION_DURATION/60} minutes to submit."
     %div{:id => "problem-submission-form-#{problem.id}"}