Fix test script, wtf!