diff --git a/test/spec/MessagesSpec.js b/test/spec/MessagesSpec.js
index 33255c1..76ad5d6 100644
--- a/test/spec/MessagesSpec.js
+++ b/test/spec/MessagesSpec.js
@@ -84,7 +84,7 @@ describe('Messages should work', function() {
     expect(send_shape, "Can't find Send Task").to.exist;
 
     // THEN - there are correlations.
-    let correlations = findEntry('messageCorrelations', container);
+    let correlations = findGroupEntry('messageCorrelations', container);
     expect(correlations, "Can't find the message correlations").to.exist;
 
     console.log("Message Correlations: ");