Fixed broken script
Fixed broken script