diff --git a/tests/forms/FieldSetTest.php b/tests/forms/FieldSetTest.php
index c5722c34b..70914ab7c 100644
--- a/tests/forms/FieldSetTest.php
+++ b/tests/forms/FieldSetTest.php
@@ -351,11 +351,7 @@ class FieldSetTest extends SapphireTest {
 			new TextField('NewField2')
 		), 'B');
 
-		$result = array();
-		foreach($fields->dataFields() as $field) {
-			$result[] = $field->Name();
-		}
-		$this->assertEquals($result, array(
+		$this->assertEquals(array_keys($fields->dataFields()), array(
 			'A',
 			'NewField1',
 			'NewField2',