diff --git a/layouts/shortcodes/wrapper.html b/layouts/shortcodes/wrapper.html
index 4b1653e..7c5965b 100644
--- a/layouts/shortcodes/wrapper.html
+++ b/layouts/shortcodes/wrapper.html
@@ -1 +1,3 @@
-<div class="{{ .Get 0 }}">{{ if eq (.Get 1) "nomarkdown" }}{{ .Inner }}{{ else }}{{ .Inner | $.Page.RenderString (dict "display" "block") }}{{ end }}</div>
+{{ $display := "inline" -}}
+{{ if eq (.Get 1) "block" }}{{ $display = "block" }}{{ end -}}
+<div class="{{ .Get 0 }}">{{ if eq (.Get 1) "nomarkdown" }}{{ .Inner }}{{ else }}{{ .Inner | $.Page.RenderString (dict "display" $display) }}{{ end }}</div>