Fix tests for CI

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