Fix test script ugh.