refactoring
minor change