minor code clean-up
2 files changed