Fix bug step 6