Removed commented code of 3d torch fix (again)