diff --git a/course/page/code_runpy_backend.py b/course/page/code_runpy_backend.py
index a06b45fc4be49a50db600dbc5accae5b82ae2e17..ded583c45b0d97ca4af560d80d428093e11462a6 100644
--- a/course/page/code_runpy_backend.py
+++ b/course/page/code_runpy_backend.py
@@ -224,7 +224,7 @@ def run_code(result, run_req):
     feedback = Feedback()
     maint_ctx = {
             "feedback": feedback,
-            "user_code": user_code,
+            "user_code": run_req.user_code,
             "data_files": data_files,
             "output_html": output_html,
             "GradingComplete": GradingComplete,