add waits in test case