Improvements to the rate-limit test script to decrease flakiness