I changed this code directly on github