[Correction] Fix the bug that requires 2 picks with the improve pick