fix shape of probe data