Wisp and AI problem.

[Krakenn99]

Hello. I have edited the Wisps with Goblin models and made them harvest Lumber like a Ghoul and Gold like a normal Wisp.

But there is the problem. When they are being attacked the workers attack back and after the attack they do not go to work again.

I also tried to completely make them like an Orc/Human worker but then the AI uses them only for Lumber and creating buildings.

Can somebody help me solve this problem?