Modified the test script to adopt changes in the load generator