Fix broken script