[fix] Upgrade script
Merge pull request #11 from Kloadut/patch-4

[fix] Upgrade script