Fix stress unit tests
Fix stress unit tests