fix baby step 2