Fixed cooldown not setting bug