Update client to new API