Fixed issue with unit prefab