fix test behaviour