diff --git a/tests/extras/extra_spec.lua b/tests/extras/extra_spec.lua
index 3bd27d34..39c3e0c0 100644
--- a/tests/extras/extra_spec.lua
+++ b/tests/extras/extra_spec.lua
@@ -27,10 +27,10 @@ describe("Extra", function()
   end, extras)
 
   local lsp_to_pkg = {}
-  if vim.fn.has("nvim-0.11") == 1 then
-    lsp_to_pkg = require("mason-lspconfig.mappings").get_mason_map().lspconfig_to_package
-  else
+  if require("lazy.core.config").spec.plugins["mason-lspconfig.nvim"].version == "^1.0.0" then
     lsp_to_pkg = require("mason-lspconfig.mappings.server").lspconfig_to_package
+  else
+    lsp_to_pkg = require("mason-lspconfig.mappings").get_mason_map().lspconfig_to_package
   end
 
   local tsspec = Plugin.Spec.new({