diff --git a/lua/telescope/builtin/__files.lua b/lua/telescope/builtin/__files.lua
index 4a1bc67..af1a65c 100644
--- a/lua/telescope/builtin/__files.lua
+++ b/lua/telescope/builtin/__files.lua
@@ -482,7 +482,7 @@ files.current_buffer_fuzzy_find = function(opts)
   end
 
   opts.results_ts_highlight = vim.F.if_nil(opts.results_ts_highlight, true)
-  local lang = vim.treesitter.language.get_lang(filetype)
+  local lang = vim.treesitter.language.get_lang(filetype) or filetype
   if opts.results_ts_highlight and lang and utils.has_ts_parser(lang) then
     local parser = vim.treesitter.get_parser(opts.bufnr, lang)
     local query = vim.treesitter.query.get(lang, "highlights")
diff --git a/lua/telescope/previewers/utils.lua b/lua/telescope/previewers/utils.lua
index ba4e444..a13ee57 100644
--- a/lua/telescope/previewers/utils.lua
+++ b/lua/telescope/previewers/utils.lua
@@ -167,7 +167,7 @@ end
 -- Attach ts highlighter
 utils.ts_highlighter = function(bufnr, ft)
   if has_filetype(ft) then
-    local lang = vim.treesitter.language.get_lang(ft)
+    local lang = vim.treesitter.language.get_lang(ft) or ft
     if lang and ts_utils.has_ts_parser(lang) then
       return vim.treesitter.start(bufnr, lang)
     end