diff --git a/init.lua b/init.lua
index a1000192..79ace422 100644
--- a/init.lua
+++ b/init.lua
@@ -165,11 +165,10 @@ require('lazy').setup({
   --
 
   {
-    'jeangiraldoo/codedocs.nvim',
-    -- Remove the 'dependencies' section if you don't plan on using nvim-treesitter
-    dependencies = {
-      'nvim-treesitter/nvim-treesitter',
-    },
+    'danymat/neogen',
+    config = true,
+    -- Uncomment next line if you want to follow only stable versions
+    -- version = "*"
   },
   -- Alternatively, use `config = function() ... end` for full control over the configuration.
   -- If you prefer to call `setup` explicitly, use:
@@ -366,7 +365,10 @@ require('lazy').setup({
         })
       end, { desc = '[/] Fuzzily search in current buffer' })
 
-      vim.keymap.set('n', '<leader>k', require('codedocs').insert_docs, { desc = 'Insert docstring' })
+      vim.keymap.set('n', '<leader>k', require('neogen').generate, {
+        noremap = true,
+        silent = true,
+      })
       -- It's also possible to pass additional configuration options.
       --  See `:help telescope.builtin.live_grep()` for information about particular keys
       vim.keymap.set('n', '<leader>s/', function()