[fix] Unit test and lint error