Fix unit tests
Signed-off-by: Dmitriy Pavlenko <dmitriy.pavlenko@sysgears.com>