diff --git a/app/sidekiq/training_job_monitor_job.rb b/app/sidekiq/training_job_monitor_job.rb
index 11aadb7..9cbb1ce 100644
--- a/app/sidekiq/training_job_monitor_job.rb
+++ b/app/sidekiq/training_job_monitor_job.rb
@@ -36,7 +36,7 @@ def perform(training_job_id, context_id)
       PredictionManifestExportJob.perform_async(context.id)
     else
       # reschedule this job to run again in 1 minute
-      TrainingJobMonitorJob.perform_in(MONITOR_JOB_RESCHEDULE_DELAY.minute, training_job.id)
+      TrainingJobMonitorJob.perform_in(MONITOR_JOB_RESCHEDULE_DELAY.minute, training_job.id, context.id)
     end
   end
 end
diff --git a/spec/sidekiq/training_job_monitor_job_spec.rb b/spec/sidekiq/training_job_monitor_job_spec.rb
index 6f6092a..ff67406 100644
--- a/spec/sidekiq/training_job_monitor_job_spec.rb
+++ b/spec/sidekiq/training_job_monitor_job_spec.rb
@@ -34,7 +34,7 @@
       allow(training_job_monitor_result).to receive(:completed?).and_return(false)
       allow(described_class).to receive(:perform_in)
       job.perform(training_job.id, context.id)
-      expect(described_class).to have_received(:perform_in).with(1.minute, training_job_monitor_result.id)
+      expect(described_class).to have_received(:perform_in).with(1.minute, training_job_monitor_result.id, context.id)
     end
 
     context 'when the monitor job returns a completed job' do