Simplify test script