diff --git a/styles/pygments.go b/styles/pygments.go
index 70d1f2a..327033b 100644
--- a/styles/pygments.go
+++ b/styles/pygments.go
@@ -5,7 +5,7 @@ import (
 )
 
 // Pygments default theme.
-var Pygments = Register(chroma.MustNewStyle("pygments", map[chroma.TokenType]string{
+var Pygments = Register(chroma.MustNewStyle("pygments", chroma.StyleEntries{
 	chroma.Whitespace:     "#bbbbbb",
 	chroma.Comment:        "italic #408080",
 	chroma.CommentPreproc: "noitalic #BC7A00",
diff --git a/styles/swapoff.go b/styles/swapoff.go
index 0971378..e4daae6 100644
--- a/styles/swapoff.go
+++ b/styles/swapoff.go
@@ -5,7 +5,7 @@ import (
 )
 
 // SwapOff theme.
-var SwapOff = Register(chroma.MustNewStyle("swapoff", map[chroma.TokenType]string{
+var SwapOff = Register(chroma.MustNewStyle("swapoff", chroma.StyleEntries{
 	chroma.Background:        "#lightgray bg:#black",
 	chroma.Number:            "bold #ansiyellow",
 	chroma.Comment:           "#ansiteal",