Applied suggestion made on the pull request.