[fix] make tests pass