Refine the test script a bit.