Minor code cleanup.
2 files changed