fixed a test

Signed-off-by: antonleykin <antonleykin@gmail.com>