Improve rate limiting logic.