diff --git a/plugins/cuda/cuda_plugin.c b/plugins/cuda/cuda_plugin.c
index e78828b18..976ce824c 100644
--- a/plugins/cuda/cuda_plugin.c
+++ b/plugins/cuda/cuda_plugin.c
@@ -391,14 +391,14 @@ int cuda_plugin_checkpoint_devices(int pid)
 	if (resume_restore_thread(restore_tid, &save_sigset)) {
 		return -1;
 	}
+
+	task_info->checkpointed = 1;
 	status = cuda_process_checkpoint_action(pid, ACTION_CHECKPOINT, 0, msg_buf, sizeof(msg_buf));
 	if (status) {
 		pr_err("CHECKPOINT_DEVICES failed with %s\n", msg_buf);
 		goto interrupt;
 	}
 
-	task_info->checkpointed = 1;
-
 interrupt:
 	int_ret = interrupt_restore_thread(restore_tid, &save_sigset);