diff --git a/htmgo-site/pages/examples/code.go b/htmgo-site/pages/examples/code.go
index 88a9364..b06f854 100644
--- a/htmgo-site/pages/examples/code.go
+++ b/htmgo-site/pages/examples/code.go
@@ -65,5 +65,10 @@ func renderCodeToString(snippet *Snippet) *h.Element {
 		source = out.String()
 	}
 
-	return ui.GoCodeSnippet(source, "border-radius: 0.5rem;")
+	return ui.CodeSnippet(ui.CodeSnippetProps{
+		Code:         source,
+		Lang:         "go",
+		CustomStyles: []string{"border-radius: 0.5rem;"},
+		WrapLines:    true,
+	})
 }
diff --git a/htmgo-site/ui/snippet.go b/htmgo-site/ui/snippet.go
index d42302e..c5e07f2 100644
--- a/htmgo-site/ui/snippet.go
+++ b/htmgo-site/ui/snippet.go
@@ -24,6 +24,7 @@ func FormatCode(props CodeSnippetProps) string {
 	lexer := lexers.Get(props.Lang)
 	style := styles.Get("github")
 	formatter := html.New(
+		html.WrapLongLines(props.WrapLines),
 		html.WithLineNumbers(!props.SingleLine && !props.HideLineNumbers),
 		html.WithCustomCSS(map[chroma.TokenType]string{
 			chroma.PreWrapper: fmt.Sprintf("border-radius: 0.2rem; line-height: 24px; font-size: 14px; padding: 12px; overflow: auto; background-color: rgb(245, 245, 245) !important; %s", strings.Join(props.CustomStyles, ";")),
@@ -42,6 +43,7 @@ type CodeSnippetProps struct {
 	CustomStyles    []string
 	HideLineNumbers bool
 	SingleLine      bool
+	WrapLines       bool
 }
 
 func CodeSnippet(props CodeSnippetProps) *h.Element {