diff --git a/mod/assign/locallib.php b/mod/assign/locallib.php
index b6505b18546..4a6aa93d106 100644
--- a/mod/assign/locallib.php
+++ b/mod/assign/locallib.php
@@ -3055,18 +3055,16 @@ class assign {
             if ($grade->grade !== null && $grade->grade >= 0) {
                 $data->grade = format_float($grade->grade, 2);
             }
+            if (!empty($flags->workflowstate)) {
+                $data->workflowstate = $flags->workflowstate;
+            }
+            if (!empty($flags->allocatedmarker)) {
+                $data->allocatedmarker = $flags->allocatedmarker;
+            }
         } else {
             $data = new stdClass();
             $data->grade = '';
         }
-
-        if (!empty($flags->workflowstate)) {
-            $data->workflowstate = $flags->workflowstate;
-        }
-        if (!empty($flags->allocatedmarker)) {
-            $data->allocatedmarker = $flags->allocatedmarker;
-        }
-
         // Warning if required.
         $allsubmissions = $this->get_all_submissions($userid);