Implemented correction suggested in github PR.