diff --git a/src/lib.rs b/src/lib.rs
index 64a24ec..33246c2 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -78,9 +78,7 @@ pub fn highlight(code: String, language: String, prefix: String) -> HighlightRes
   let is_plain = language == "plain" || language == "plaintext";
   let mut syntax = syntax_set.find_syntax_plain_text();
   if !is_plain {
-    match syntax_set
-      .find_syntax_by_extension(&language)
-      .or(syntax_set.find_syntax_by_name(&language))
+    match syntax_set.find_syntax_by_token&language)
     {
       Some(s) => syntax = s,
       _ => (),