Correct shape for inference algos