Build break fix.

Fix build break introduced by last API change.