Warden immediately burrows back to the ground after using spawn egg, and being name tagged

In 26.1 Snapshot 3 Wardens summoned with spawn egg and nametagged immiediately burrows back to the ground. We can notice same behavior after using summon command or spawn it with Sculk Shrieker and name it, but only after player death or rejoin the world.

How to trigger it:
1. Create new world on the previous snapshot and summon Warden with command summon, spawn egg or Sculk Shrieker and named it. Next open world on the newest snapshot and You can notice that Wardens immediately burrows back. Same thing happens while using spawn egg all the time.
2. Create new world on newest Snapshot, use summon command or spawn Warden with Sculk Shrieker and name it. Use kill command on the Player or quit the world and back again to see Warden burrowing back.

Expected result:
1. Nametagged Warden will not burrow back after version changes , Player death or rejoing their world
2. Warden spawned with their spawn egg will not burrow back

Some examples attached below.