diff --git a/form.js b/form.js
index 066689468f1ecaab0016b0c1f9d6275311edc6eb..ef6e71c5ee6f0e4915e9624b3b7233c1d5531669 100644
--- a/form.js
+++ b/form.js
@@ -36,6 +36,6 @@ function toggle_visibilty_of_form_group(form_id, show) {
 
 toggle_gpu_visibility(
   // Fake the event
-  { target: document.querySelector('#batch_connect_session_context_bc_num_gpus') }
+  { target: document.querySelector('#batch_connect_session_context_bc_partition') }
 );
 set_partition_change_handler();