minor refactor
2 files changed