API change. Missed in rebase