[Tests] minor improvement