chore: address feedback
2 files changed