diff --git a/core/graph/src/test/java/uk/gov/gchq/gaffer/graph/GraphTest.java b/core/graph/src/test/java/uk/gov/gchq/gaffer/graph/GraphTest.java
index ae42ec9d588..dd1a0111cab 100644
--- a/core/graph/src/test/java/uk/gov/gchq/gaffer/graph/GraphTest.java
+++ b/core/graph/src/test/java/uk/gov/gchq/gaffer/graph/GraphTest.java
@@ -928,8 +928,8 @@ public void shouldConstructGraphAndCreateViewWithGroups(@Mock final Store store)
 
         // Then
         assertNotSame(schema, resultView);
-        assertArrayEquals(entities.keySet().toArray(), resultView.getEntityGroups().toArray());
-        assertArrayEquals(edges.keySet().toArray(), resultView.getEdgeGroups().toArray());
+        assertThat(entities.keySet()).isEqualTo(resultView.getEntityGroups());
+        assertThat(edges.keySet()).isEqualTo(resultView.getEdgeGroups());
 
         for (final ViewElementDefinition resultElementDef : resultView.getEntities().values()) {
             assertNotNull(resultElementDef);