Fixin the unstable test for ambient tx to wait for the non tx message to
arrive before completing.