Fix recovery bug