Revamp the script to handle the 'TEST=simple' output.