Change as suggested in pull request

Thanks @gregcaporaso