Fix the script to run the tests