diff --git a/mod/data/tests/generator_test.php b/mod/data/tests/generator_test.php
index 1f443c67eee..cb199d83f74 100644
--- a/mod/data/tests/generator_test.php
+++ b/mod/data/tests/generator_test.php
@@ -207,7 +207,7 @@ class mod_data_generator_testcase extends advanced_testcase {
         $entry = $DB->get_record('data_records', array('id' => $datarecordid));
         $this->assertEquals($entry->groupid, $groupa->id);
 
-        $contents = $DB->get_records('data_content', array('recordid' => $datarecordid));
+        $contents = $DB->get_records('data_content', array('recordid' => $datarecordid), 'id');
 
         $contentstartid = 0;
         $flag = 0;