it seems like now the duration remaining on the ability gets added to the cooldown so you end up having to wait a lot longer for it to come back than you would if it expired naturally, which makes coming back way harder as now you have to survive a minute without a special ability the game difficulty is balanced around. just have it go to the time it would have if it expired naturally instead, or dont have the ability dissapear when you die