Try to fix flaky tests

[SiarheiFedartsou]

Issue

What issue is this PR targeting? If there is no issue that addresses the problem, please open a corresponding issue and link it here.

Please read our documentation on release and version management. If your PR is still work in progress please attach the relevant label.

Tasklist

• [ ] CHANGELOG.md entry (How to write a changelog entry)
• [ ] update relevant Wiki pages
• [ ] add tests (see testing documentation)
• [ ] review
• [ ] adjust for comments
• [ ] cherry pick to release branch

Requirements / Relations

Link any requirements here. Other pull requests this PR is based on?