Make rate limiting unit test more robust