Make the script compatible with new github changes
Merge pull request #2 from giftig/fix

Make the script compatible with new github changes