diff --git a/test/ejabberd_SUITE.erl b/test/ejabberd_SUITE.erl
index 47c4b9e75..a220ed37e 100644
--- a/test/ejabberd_SUITE.erl
+++ b/test/ejabberd_SUITE.erl
@@ -1688,7 +1688,8 @@ mam_query_all(Config, NS) ->
                                                body = [Text]}]}]}]})
       end, Iter),
     if NS == ?NS_MAM_TMP ->
-	    ?recv1(#iq{type = result, id = I, sub_els = []});
+	    ?recv1(#iq{type = result, id = I,
+		       sub_els = [#mam_query{xmlns = NS, id = QID}]});
        true ->
 	    ?recv1(#message{sub_els = [#mam_fin{complete = true, id = QID}]})
     end.
@@ -1724,7 +1725,8 @@ mam_query_with(Config, JID, NS) ->
                                                body = [Text]}]}]}]})
       end, Iter),
     if NS == ?NS_MAM_TMP ->
-	    ?recv1(#iq{type = result, id = I, sub_els = []});
+	    ?recv1(#iq{type = result, id = I,
+		       sub_els = [#mam_query{xmlns = NS}]});
        true ->
 	    ?recv1(#message{sub_els = [#mam_fin{complete = true}]})
     end.