Replicating root motion montage

Hello! I cannot figure out how to replicate my root motion montages properly. The issue I’m having is with my AI. The Montages are for hit reactions when getting shot, so when the AI gets shot, it stumbles back in return. However, when testing with Client, the AI will do the hit reaction animation momentarily and then immediately teleport to me as if they kept running.
This is what I’m using for the montages and the custom events to replicate it.