Enhance rate limit retry logic