diff --git a/protocol/peersyncing/peersyncing_test.go b/protocol/peersyncing/peersyncing_test.go
index 6abe992c418..7f0c6407fe7 100644
--- a/protocol/peersyncing/peersyncing_test.go
+++ b/protocol/peersyncing/peersyncing_test.go
@@ -115,10 +115,12 @@ func (s *PeerSyncingSuite) TestOrderAndLimit() {
 	byChatID, err := s.p.AvailableMessagesMapByChatIDs([][]byte{testCommunityID}, 10)
 
 	s.Require().NoError(err)
-	s.Require().Len(byChatID, 4)
+	s.Require().Len(byChatID, 1)
+	s.Require().Len(byChatID[string(testCommunityID)], 4)
 
 	byChatID, err = s.p.AvailableMessagesMapByChatIDs([][]byte{testCommunityID}, 3)
 
 	s.Require().NoError(err)
-	s.Require().Len(byChatID, 3)
+	s.Require().Len(byChatID, 1)
+	s.Require().Len(byChatID[string(testCommunityID)], 3)
 }