Fix update to return updated obj