[tests] Fix for unit test