removing wait in test