SCHEMA-143. Update to reflect new API.

Signed-off-by: Robert Chu <robert@wibidata.com>