Fixed tests

Signed-off-by: Razican <admin@razican.com>