BUG: fixed DataProbe module to work with non-linear transform