Fix pull request feedback