diff --git a/.drone.yml b/.drone.yml
index 7277aeb..f84c99f 100644
--- a/.drone.yml
+++ b/.drone.yml
@@ -26,7 +26,7 @@ steps:
       NPM_CONFIG_LOGLEVEL: error
 
   - name: testbuild
-    image: thegeeklab/hugo:0.97.3
+    image: thegeeklab/hugo:0.104.3
     commands:
       - mkdir -p exampleSite/themes/ && ln -s $(pwd)/ exampleSite/themes/hugo-geekblog
       - hugo --panicOnWarning -s exampleSite/ -b http://localhost:8000/ -e ci
@@ -206,7 +206,7 @@ steps:
         - refs/pull/**
 
   - name: build
-    image: thegeeklab/hugo:0.97.3
+    image: thegeeklab/hugo:0.104.3
     commands:
       - hugo --panicOnWarning -s exampleSite/
 
@@ -287,6 +287,6 @@ depends_on:
 
 ---
 kind: signature
-hmac: f04800844c631b0c9e17b7a7c015ca971754516f609248a380ee0db2ec2037c7
+hmac: 7124323835f4329636bb5445e8018ad366e6cbab3caec54e30edd7939603cb1a
 
 ...
diff --git a/exampleSite/content/posts/advanced/shortcodes.md b/exampleSite/content/posts/advanced/shortcodes.md
index 25924c4..0059d73 100644
--- a/exampleSite/content/posts/advanced/shortcodes.md
+++ b/exampleSite/content/posts/advanced/shortcodes.md
@@ -348,11 +348,9 @@ Boxes can be used to create a simple grid.
 
 ### Example
 
-{{< columns >}}
-
 <!-- prettier-ignore -->
 ```tpl
-{{</* mermaid class="text-center" */>}}
+{{</*/* mermaid class="text-center" */*/>}}
 sequenceDiagram
     Alice->>Bob: Hello Bob, how are you?
     alt is sick
@@ -363,11 +361,9 @@ sequenceDiagram
     opt Extra response
         Bob->>Alice: Thanks for asking
     end
-{{</* /mermaid */>}}
+{{</*/* /mermaid */*/>}}
 ```
 
-<--->
-
 <!-- spellchecker-disable -->
 <!-- prettier-ignore -->
 {{< mermaid class="text-center" >}}
@@ -383,10 +379,6 @@ sequenceDiagram
     end
 {{< /mermaid >}}
 
-<!-- spellchecker-enable -->
-
-{{< /columns >}}
-
 As an alternative to shortcodes, code blocks can be used for markdown as well.
 
 {{< columns >}}
@@ -424,16 +416,12 @@ C -->|Two| E[Result 2]
 
 ### Example
 
-{{< columns >}}
-
 ```latex
-{{</* katex [display] [class="text-center"] */>}}
+{{</*/* katex [display] [class="text-center"] */*/>}}
 f(x) = \int_{-\infty}^\infty\hat f(\xi)\,e^{2 \pi i \xi x}\,d\xi
-{{</* /katex */>}}
+{{</*/* /katex */*/>}}
 ```
 
-<--->
-
 <!-- spellchecker-disable -->
 <!-- prettier-ignore -->
 {{< katex display >}}
@@ -442,17 +430,7 @@ f(x) = \int_{-\infty}^\infty\hat f(\xi)\,e^{2 \pi i \xi x}\,d\xi
 
 <!-- spellchecker-enable -->
 
-{{< /columns >}}
-
-KaTeX can also be used inline, for example {{< katex >}}\pi(x){{< /katex >}} or used with the `display` parameter (e.g. `display: block`):
-
-<!-- spellchecker-disable -->
-<!-- prettier-ignore -->
-{{< katex display >}}
-f(x) = \int_{-\infty}^\infty\hat f(\xi)\,e^{2 \pi i \xi x}\,d\xi
-{{< /katex >}}
-
-<!-- spellchecker-enable -->
+KaTeX can be used inline, for example {{< katex >}}\pi(x){{< /katex >}} or used with the `display` parameter (see above).
 
 Text continues here.
 
diff --git a/layouts/_default/_markup/render-codeblock-mermaid.html b/layouts/_default/_markup/render-codeblock-mermaid.html
index ce65354..d17ac6f 100644
--- a/layouts/_default/_markup/render-codeblock-mermaid.html
+++ b/layouts/_default/_markup/render-codeblock-mermaid.html
@@ -6,6 +6,6 @@
 {{ end }}
 <!-- prettier-ignore-end -->
 
-<pre class="gblog-mermaid mermaid">
+<pre class="gblog-mermaid mermaid text-center">
   {{- .Inner -}}
 </pre>