added wait in test case