fixed torch WAILA compat