Fix extension test to match API change.