slight code edit
2 files changed