Fixed test script so coverage works.