diff --git a/config/nvim/lua/config/diagnostic.lua b/config/nvim/lua/config/diagnostic.lua
index 47a1be3..b94d7b4 100644
--- a/config/nvim/lua/config/diagnostic.lua
+++ b/config/nvim/lua/config/diagnostic.lua
@@ -87,14 +87,14 @@ M.set_severity = function()
         end,
       },
       sorter = require("telescope.config").values.generic_sorter(opts),
-      attach_mappings = function(prompt_bufnr, _)
-        local actions = require("telescope.actions")
-        actions.select_default:replace(function()
-          actions.close(prompt_bufnr)
+      attach_mappings = function(prompt_bufnr)
+        require("telescope.actions.set").select:replace(function()
           local selection = require("telescope.actions.state").get_selected_entry()
+          require("telescope.actions").close(prompt_bufnr)
+
           M.setup { severity = { min = selection.value.severity } }
         end)
-        return true
+        return true -- Attach default mappings as well.
       end,
     })
     :find()