Fix cooldown not always working