minor fix
refactoring