(fix) Correct step4.jsx

(fix) Refactor step4.jsx