cooldown code fix