[Support][Test] Missed this in the API change.