diff --git a/hugolib/shortcode_test.go b/hugolib/shortcode_test.go
index 2ac525101..54abdecd9 100644
--- a/hugolib/shortcode_test.go
+++ b/hugolib/shortcode_test.go
@@ -599,7 +599,7 @@ tags:
 		} else if strings.HasSuffix(test.contentPath, ".rst") && !helpers.HasRst() {
 			fmt.Println("Skip Rst test case as no rst2html present.")
 			continue
-		} else if strings.Contains(test.expected, "code class") && !helpers.HasPygments() {
+		} else if strings.Contains(test.expected, "code") && !helpers.HasPygments() {
 			fmt.Println("Skip Pygments test case as no pygments present.")
 			continue
 		}