minor refactoring
minor fix