Totem of Undying death animation model doesn't match with CustomModelData

Steps to reproduce

• Download the test resource pack.
• Use 

  /give @s totem_of_undying{CustomModelData:1}

• Use

  /tp ~ ~50 ~

• When totem death animation has displayed, the model still rendered original texture.