fix a minor refactoring issue