fix test to match behaviour