Fix rate-limiting tests