From 95244c3b6fbb13f64ae100b3cb1e6773e21ffdbe Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Pawe=C5=82=20Chmielowski?= <pchmielowski@process-one.net>
Date: Thu, 12 Apr 2018 17:08:27 +0200
Subject: [PATCH] Fix csi tests

---
 test/csi_tests.erl | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/test/csi_tests.erl b/test/csi_tests.erl
index f9d47d55f..2b346e275 100644
--- a/test/csi_tests.erl
+++ b/test/csi_tests.erl
@@ -55,7 +55,7 @@ master_slave_cases() ->
 all_master(Config) ->
     Peer = ?config(peer, Config),
     Presence = #presence{to = Peer},
-    ChatState = #message{to = Peer, thread = <<"1">>,
+    ChatState = #message{to = Peer, thread = #message_thread{data = <<"1">>},
 			 sub_els = [#chatstate{type = active}]},
     Message = ChatState#message{body = [#text{data = <<"body">>}]},
     PepPayload = xmpp:encode(#presence{}),
@@ -133,15 +133,15 @@ all_slave(Config) ->
 			  [#ps_item{
 			      id = <<"pep-2">>}]}},
 	    #delay{}]} = recv_message(Config),
-    #message{from = Peer, thread = <<"1">>,
+    #message{from = Peer, thread = #message_thread{data = <<"1">>},
 	     sub_els = [#chatstate{type = composing},
 			#delay{}]} = recv_message(Config),
-    #message{from = Peer, thread = <<"1">>,
+    #message{from = Peer, thread = #message_thread{data = <<"1">>},
 	     body = [#text{data = <<"body">>}],
 	     sub_els = [#chatstate{type = active}]} = recv_message(Config),
     change_client_state(Config, active),
     wait_for_master(Config),
-    #message{from = Peer, thread = <<"1">>,
+    #message{from = Peer, thread = #message_thread{data = <<"1">>},
 	     sub_els = [#chatstate{type = active}]} = recv_message(Config),
     disconnect(Config).