diff --git a/share/jupyter/nbconvert/templates/lab/base.html.j2 b/share/jupyter/nbconvert/templates/lab/base.html.j2
index 45046c599..f7998f942 100644
--- a/share/jupyter/nbconvert/templates/lab/base.html.j2
+++ b/share/jupyter/nbconvert/templates/lab/base.html.j2
@@ -86,6 +86,7 @@
 
 {% block markdowncell scoped %}
 <div class="jp-Cell-inputWrapper">
+<div class="jp-InputArea jp-Cell-inputArea">
 {%- if resources.global_content_filter.include_input_prompt-%}
     {{ self.empty_in_prompt() }}
 {%- endif -%}
@@ -93,6 +94,7 @@
 {{ cell.source  | markdown2html | strip_files_prefix }}
 </div>
 </div>
+</div>
 {%- endblock markdowncell %}
 
 {% block unknowncell scoped %}