[tests] Fixed tests that were weirdly failing (#2908).

[tests] Fixed tests that were weirdly failing (#2908). #32