Fix missing cooldown bug

It now works