Set proper wait time