small fix
refactoring