Fix the test running script