Update SampleSpatialization to match new API