fix step 1 bug