refactor
minor fix