From 78dae4036e93e0d08a2052ef7e9a8b7f7726421e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Pawe=C5=82=20Chmielowski?= <pchmielowski@process-one.net>
Date: Mon, 17 Sep 2018 09:34:15 +0200
Subject: [PATCH] Reintroduce change removed by mistake in 'Improve match
 macro'

---
 test/suite.erl | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/test/suite.erl b/test/suite.erl
index efe587253..ecdbd71fb 100644
--- a/test/suite.erl
+++ b/test/suite.erl
@@ -471,7 +471,9 @@ wait_auth_SASL_result(Config, ShouldFail) ->
 	    NS = if Type == client -> ?NS_CLIENT;
 		    Type == server -> ?NS_SERVER
 		 end,
-	    receive #stream_start{xmlns = NS, version = {1,0}} -> ok end,
+	    Config2 = receive #stream_start{id = ID, xmlns = NS, version = {1,0}} ->
+		set_opt(stream_id, ID, Config)
+	    end,
             receive #stream_features{sub_els = Fs} ->
 		    if Type == client ->
 			    #xmpp_session{optional = true} =
@@ -488,7 +490,7 @@ wait_auth_SASL_result(Config, ShouldFail) ->
 			      set_opt(rosterver, true, ConfigAcc);
 			 (_, ConfigAcc) ->
 			      ConfigAcc
-		      end, Config, Fs)
+		      end, Config2, Fs)
 	    end;
         #sasl_challenge{text = ClientIn} ->
             {Response, SASL} = (?config(sasl, Config))(ClientIn),