Fixed broken script