minor change

code refactoring.