Close #22, Close #23

1 files changed, 8 insertions, 5 deletions

diff --git a/src/clarktown/renderers/heading_block.clj b/src/clarktown/renderers/heading_block.clj
index 4da9bda..7f4c7de 100644
--- a/src/clarktown/renderers/heading_block.clj
+++ b/src/clarktown/renderers/heading_block.clj
@@ -7,8 +7,7 @@
 (defn render-atx-heading
   "Renders the hashbang heading block."
   [block]
-  (let [single-line-block (-> (string/replace block #"\n" "")
-                              string/trim)
+  (let [single-line-block (string/trim block)
         size (-> (string/split single-line-block #" ")
                  first
                  string/trim
@@ -39,6 +38,10 @@
 (defn render
   "Renders the heading block."
   [block _ _]
-  (if (matcher/is-atx-heading? block)
-    (render-atx-heading block)
-    (render-settext-heading block)))
+  (cond (matcher/is-atx-heading? block)
+        (render-atx-heading block)
+        
+        (matcher/is-settext-heading? block)
+        (render-settext-heading block)
+        
+        :else block))