Try to fix flapping test case