Fixing broken test script