diff --git a/scripts/pygmentize_dir.py b/scripts/pygmentize_dir.py
index 4c79802..8f189e2 100755
--- a/scripts/pygmentize_dir.py
+++ b/scripts/pygmentize_dir.py
@@ -1,10 +1,10 @@
-#!/usr/bin/env python
+#!/usr/bin/env python3
 
 import os
 import re
 from glob import glob
 
-from pygments.formatters.html import HtmlFormatter
+from pygments.formatters import HtmlFormatter
 from pygments import highlight
 
 from cddl_lexer import CustomLexer as CddlLexer