small fix for cooldown animation