diff --git a/mod/assign/locallib.php b/mod/assign/locallib.php
index ed5f3ed5f7c..46404bc7eb3 100644
--- a/mod/assign/locallib.php
+++ b/mod/assign/locallib.php
@@ -1888,12 +1888,12 @@ class assign {
         if (!$batchusers) {
             $userid = required_param('userid', PARAM_INT);
 
-            $grade = $this->get_user_grade($userid, false);
+            $flags = $this->get_user_flags($userid, false);
 
             $user = $DB->get_record('user', array('id'=>$userid), '*', MUST_EXIST);
 
-            if ($grade) {
-                $data->extensionduedate = $grade->extensionduedate;
+            if ($flags) {
+                $data->extensionduedate = $flags->extensionduedate;
             }
             $data->userid = $userid;
         } else {