Fixed file based on pull request comments.