add fix from https://github.com/mordyovits to original repo