[minor] Address justin's feedback on API changes pr