Added missing wait time in test