diff --git a/src/jupyter_contrib_nbextensions/nbextensions/python-markdown/main.js b/src/jupyter_contrib_nbextensions/nbextensions/python-markdown/main.js
index aab02df..724d61e 100644
--- a/src/jupyter_contrib_nbextensions/nbextensions/python-markdown/main.js
+++ b/src/jupyter_contrib_nbextensions/nbextensions/python-markdown/main.js
@@ -63,9 +63,9 @@ define([
                             html = marked(text);
                         } else if (out_data.msg_type === "stream") {
                             html = marked(out_data.content.text);
-                            var t = html.match(/<p>([\s\S]*?)<\/p>/)[1]; //strip <p> and </p> that marked adds and we don't want
-                            html = t ? t : html;
-                            var q = html.match(/&#39;([\s\S]*?)&#39;/); // strip quotes from strings
+                            var t = html.match(/^\s*<p>([\s\S]*?)<\/p>\s*$/); //strip <p> and </p> that marked (maybe) adds and we don't want
+                            html = t !== null ? t[1] : html;
+                            var q = html.match(/^&#39;([\s\S]*?)&#39;$/); // strip quotes from strings
                             if (q !== null) html = q[1]
                         } else if (out_data.msg_type === "execute_result" | out_data.msg_type === "display_data" ) {
                             var ul = out_data.content.data;
@@ -90,9 +90,9 @@ define([
                                 } else {
                                     html = marked(ul['text/plain']);
                                     // [\s\S] is used to also catch newlines
-                                    var t = html.match(/<p>([\s\S]*?)<\/p>/)[1]; //strip <p> and </p> that marked adds and we don't want
-                                    html = t ? t : html;
-                                    var q = html.match(/&#39;([\s\S]*?)&#39;/); // strip quotes from strings
+                                    var t = html.match(/^\s*<p>([\s\S]*?)<\/p>\s*$/); //strip <p> and </p> that marked adds and we don't want
+                                    html = t !== null ? t[1] : html;
+                                    var q = html.match(/^&#39;([\s\S]*?)&#39;$/); // strip quotes from strings
                                     if (q !== null) html = q[1]
                                 }
                             }