[fix] unit tests