Fix small bug per pull request on github. Thanks staticglobal.