diff --git a/cms/djangoapps/contentstore/views.py b/cms/djangoapps/contentstore/views.py
index b4889b0621..609f800951 100644
--- a/cms/djangoapps/contentstore/views.py
+++ b/cms/djangoapps/contentstore/views.py
@@ -222,7 +222,7 @@ def edit_unit(request, location):
 
     return render_to_response('unit.html', {
         'unit': item,
-        'unit_location': published_location,
+        'unit_location': location,
         'components': components,
         'component_templates': component_templates,
         'draft_preview_link': lms_link,