Using setCooldown for the shield while the player is blocking with it won't stop the using of the shield. This could be a Bug - or not expected to happen.

As solution for this (and just because nice to have) there is the Method:
LivingEntity#stopUsingItem