Fix random failing test

Signed-off-by: Dmitriy Zaporozhets <dmitriy.zaporozhets@gmail.com>