Fix missing pygments highlighting