Fixed up the step 2.2