Stun effect, cancel entity from moving

[LuccaPossamai]

I'm trying to make an stun effect that don't let entity move. In the created Effect Class I write this lines with the "LivingEvent.LivingUpdateEvent", but when the entity have the Effect, it don't take damage, don't have gravity and other limitations. Is there some way to cancel only the move from it?

@SubscribeEvent
    public static void dontMove(LivingEvent.LivingUpdateEvent event){
        LivingEntity livingEntity = event.getEntityLiving();
        if (livingEntity.hasEffect(stunningInstance)) {
            event.setCanceled(true);
        }
    }

 

[poopoodice]

You can replace their AI tasks with your own one.

[LuccaPossamai]

With the AI entities, yes, but with the Stunned players?