Revert this to what's in master
Updating syntax to match new api