fixes test script (hopefully)