* fix build * fix tests * update * fix * update * Bump Co-authored-by: Bryan Chen <[email protected]>