Fix tests timeout

[darangi]

Please be sure to read the contributor's guide before submitting any pull requests.

Requirements

• Filling out the template is required. Any pull request that does not include enough information to be reviewed in a timely manner may be closed at the maintainers' discretion.
• All new code requires tests to ensure against regressions.

Description of the Change

Alternate Designs

Benefits

Possible Drawbacks

Verification Process

Applicable Issues