* fix build * fix tests * update * fix * update * Bump Co-authored-by: Bryan Chen <xlchen1291@gmail.com>