Make the test case more realiable