Model not showing after being trained

I’ve been trying to give the tempest the tal’darim model which works fine in the editor and in game, if its already on the map. but whenever it is trained from the stargate it uses the default tempest model instead, this also only happens with ships as units and vehicles work fine when they are trained.

Sorta figured it out, instead of basing the train command for the building on the stargate I based it on the gateway instead and it worked fine.