Fix current tests

Signed-off-by: Kirill Enykeev <enykeev@gmail.com>