diff --git a/mod/assign/tests/generator/lib.php b/mod/assign/tests/generator/lib.php
index 6c225819bfa..9830af7045d 100644
--- a/mod/assign/tests/generator/lib.php
+++ b/mod/assign/tests/generator/lib.php
@@ -76,6 +76,7 @@ class mod_assign_generator extends testing_module_generator {
 
         $record->coursemodule = $this->precreate_course_module($record->course, $options);
         $id = assign_add_instance($record, null);
+        rebuild_course_cache($record->course, true);
         return $this->post_add_instance($id, $record->coursemodule);
     }
 }
diff --git a/mod/assign/tests/lib_test.php b/mod/assign/tests/lib_test.php
index 420a22f17ea..304adc2f0dd 100644
--- a/mod/assign/tests/lib_test.php
+++ b/mod/assign/tests/lib_test.php
@@ -40,23 +40,26 @@ require_once($CFG->dirroot . '/mod/assign/tests/base_test.php');
 class mod_assign_lib_testcase extends mod_assign_base_testcase {
 
     public function test_assign_print_overview() {
+        global $DB;
         $this->setUser($this->editingteachers[0]);
         $this->create_instance();
         $this->create_instance(array('duedate'=>time()));
 
+        $courses = $DB->get_records('course', array('id' => $this->course->id));
+
         $this->setUser($this->students[0]);
         $overview = array();
-        assign_print_overview(array($this->course->id => $this->course), $overview);
+        assign_print_overview($courses, $overview);
         $this->assertEquals(count($overview), 1);
 
         $this->setUser($this->teachers[0]);
         $overview = array();
-        assign_print_overview(array($this->course->id => $this->course), $overview);
+        assign_print_overview($courses, $overview);
         $this->assertEquals(count($overview), 1);
 
         $this->setUser($this->editingteachers[0]);
         $overview = array();
-        assign_print_overview(array($this->course->id => $this->course), $overview);
+        assign_print_overview($courses, $overview);
         $this->assertEquals(1, count($overview));
     }