diff --git a/cms/templates/unit.html b/cms/templates/unit.html
index 3777c33591..84785a362d 100644
--- a/cms/templates/unit.html
+++ b/cms/templates/unit.html
@@ -64,15 +64,17 @@
% for type, templates in sorted(component_templates.items()):
-
-
+ % if type == "problem":
+
+
+ % endif
% for name, location, has_markdown, is_empty in templates:
@@ -92,13 +94,13 @@
% endif
% endif
-
+
%endfor
-
-
- % if type == "problem":
+ % if type == "problem":
+
+
% for name, location, has_markdown, is_empty in templates:
% if not has_markdown:
% if is_empty:
@@ -118,10 +120,10 @@
% endif
% endif
% endfor
- % endif
-
-
+
+
+ % endif
Cancel
% endfor