fix back_to_step1 function