diff --git a/neovim/plugins/diagram-nvim.nix b/neovim/plugins/diagram-nvim.nix
index 19987cf..86a010e 100644
--- a/neovim/plugins/diagram-nvim.nix
+++ b/neovim/plugins/diagram-nvim.nix
@@ -1,15 +1,12 @@
 {
   lib,
-  helpers,
-  config,
-  pkgs,
   ...
 }:
-helpers.neovim-plugin.mkNeovimPlugin config {
+lib.nixvim.neovim-plugin.mkNeovimPlugin {
   name = "diagram-nvim";
   originalName = "diagram.nvim";
   luaName = "diagram";
-  defaultPackage = pkgs.vimPlugins.diagram-nvim;
+  package = "diagram-nvim";
 
   maintainers = [ lib.maintainers.traxys ];
 }