diff --git a/core/lib/dal/doc/ProofGenerationDal.md b/core/lib/dal/doc/ProofGenerationDal.md
index 618fdfba13b..40ee31a4b1a 100644
--- a/core/lib/dal/doc/ProofGenerationDal.md
+++ b/core/lib/dal/doc/ProofGenerationDal.md
@@ -11,9 +11,10 @@ proof_generation_details
 title: Status Diagram
 ---
 stateDiagram-v2
-[*] --> ready_to_be_proven : insert_proof_generation_details
-ready_to_be_proven --> picked_by_prover : get_next_block_to_be_proven
+[*] --> unpicked : insert_proof_generation_details
+unpicked --> picked_by_prover : lock_batch_for_proving
 picked_by_prover --> generated : save_proof_artifacts_metadata
+picked_by_prover --> unpicked : unlock_batch
 generated --> [*]
 
 [*] --> skipped : mark_proof_generation_job_as_skipped