Test: fix test