Try giving a look into this. I'm not too sure it's there but better than nothing i guess.
Setting the shock duration to like 1 second could also work out for you, way simpler to do as well.
Edit History
Please note that revisions older than 365 days are pruned and will no longer show here
Try giving a look into this. I'm not too sure it's there but better than nothing i guess