Now fix OmniTorch.