UPD: fix test