fix OmniTorch