Major improvement in TensorField.restrict