minor fix
refactor