[fix] Fix  code
[new] Test for