fix the remaining bug of step 3