OmniTorch: fix.  Need to fix for real later