Add helper test with unit test and fix this feature