Update tests to reflect new behavior
Update Push to reflect the correct API