Test method which invokes another method with runAsync() block

I would be grateful for any advice or practices which you follow.

We use Mockito and Junit4/5 to to write unit tests. However, I see that some tests fail at prod (when at the end of a test you verify calling of methods which have runAsync block).

Are there exists any options to fix these tests except adding timeout to verify or excluding runAsync() blocks? TQ in advance.