Update test script to be like production