Fix broken test script