Работники после ополчения не строят здания

Написал триггер для кастомной расы, где вместо работников с ратушей спавняться другие юниты. У новых рабочих та же способность, что и у обычных переходить в ополчение. Я сделал специально для них новую способность на основе обычного ополчения, но при выходе из него они больше не строят кастомные здания, а оригинальные да. Я просматривал другие карты с похожей механикой и триггерами, делал также как там, но ничего не помогло.