small refactoring
small refactoring fix