Fixed test script works on all inputs