fix mistake
refactor