Custom mob

Hi, I created a custom mob using AJ in block bench and I use MC stacker to generate an invisible wither skeleton that be the AI for my custom mob. But what happened in my world is when the wither skeleton die the custom mob model is still there standing and I have to run a function to remove it. How do I make it when the wither skeleton die the model delete itself?