Replace some receive after in tests to test timeout