Update for api changes

Updating for:
justinabrahms/imhotep#31