add cooldown back